diff --git a/pkg/tests/test_db_adapter.py b/pkg/tests/test_db_adapter.py
index 340e6fb..00c67ca 100644
--- a/pkg/tests/test_db_adapter.py
+++ b/pkg/tests/test_db_adapter.py
@@ -9,7 +9,8 @@
 have_data_algebra = False
 try:
     import data_algebra
-    from data_algebra.data_ops import *
+    from data_algebra.view_representations import TableDescription
+    from data_algebra.data_ops import descr, describe_table
     import data_algebra.SQLite
     from vtreat.vtreat_db_adapter import as_data_algebra_pipeline
     have_data_algebra = True
diff --git a/pkg/vtreat/__init__.py b/pkg/vtreat/__init__.py
index 7460420..5905ae9 100644
--- a/pkg/vtreat/__init__.py
+++ b/pkg/vtreat/__init__.py
@@ -8,7 +8,8 @@
 # noinspection PyUnresolvedReferences
 import numpy
 
-from vtreat.vtreat_api import *
+from vtreat.vtreat_api import unsupervised_parameters, vtreat_parameters, BinomialOutcomeTreatment, MultinomialOutcomeTreatment, NumericOutcomeTreatment, UnsupervisedTreatment
+
 
 __docformat__ = "restructuredtext"
 __version__ = "1.3.0"