diff --git a/examples/demo_find_hyperparams.py b/examples/demo_find_hyperparams.py
index a3c2550..8b06055 100644
--- a/examples/demo_find_hyperparams.py
+++ b/examples/demo_find_hyperparams.py
@@ -1,7 +1,7 @@
 from sklearn.datasets import load_iris
 from sklearn.model_selection import train_test_split
 
-from pylmnn import find_hyperparams
+from pylmnn.bayesopt import find_hyperparams
 
 
 X, y = load_iris(return_X_y=True)
diff --git a/pylmnn/__init__.py b/pylmnn/__init__.py
index 18f7b21..1e75ee1 100755
--- a/pylmnn/__init__.py
+++ b/pylmnn/__init__.py
@@ -1,2 +1 @@
 from .lmnn import LargeMarginNearestNeighbor, make_lmnn_pipeline
-from .bayesopt import find_hyperparams
\ No newline at end of file
diff --git a/setup.py b/setup.py
index c8e186a..f833443 100755
--- a/setup.py
+++ b/setup.py
@@ -4,7 +4,7 @@
 from codecs import open
 from os import path
 
-__version__ = '1.6.1'
+__version__ = '1.6.2'
 
 if __name__ == '__main__':
     here = path.abspath(path.dirname(__file__))