diff --git a/lime/discretize.py b/lime/discretize.py
index 59fd4687..9013f412 100644
--- a/lime/discretize.py
+++ b/lime/discretize.py
@@ -3,6 +3,7 @@
 """
 import numpy as np
 import sklearn
+import sklearn.tree
 from abc import ABCMeta, abstractmethod
 
 
diff --git a/setup.py b/setup.py
index 1e029370..b678999c 100644
--- a/setup.py
+++ b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import setup
 
 setup(name='lime',
-      version='0.1.1.23',
+      version='0.1.1.24',
       description='Local Interpretable Model-Agnostic Explanations for machine learning classifiers',
       url='http://github.com/marcotcr/lime',
       author='Marco Tulio Ribeiro',