diff --git a/estimators/basic-usage.py b/estimators/basic-usage.py
index 51221ac..41dace9 100644
--- a/estimators/basic-usage.py
+++ b/estimators/basic-usage.py
@@ -1,9 +1,9 @@
 import argparse, os, gzip
-import cressieread
-import ips_snips
-import mle
+from contextual_bandits import cressieread
+from contextual_bandits import ips_snips
+from contextual_bandits import mle
 import ds_parse
-import cats_utils
+from contextual_bandits import cats_utils
 
 
 def compute_estimates(log_fp, cats_transformer=None):
diff --git a/estimators/test/__init__.py b/estimators/test/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/estimators/test/test_pi.py b/estimators/test/test_pi.py
index f9d3838..1fe09e5 100644
--- a/estimators/test/test_pi.py
+++ b/estimators/test/test_pi.py
@@ -1,9 +1,9 @@
 import os, sys
 sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
-import pseudo_inverse
-import ips_snips
-import cats_utils
+from slates import pseudo_inverse
+from contextual_bandits import ips_snips
+from contextual_bandits import cats_utils
 
 def test_single_slot_pi_equivalent_to_ips():
     """PI should be equivalent to IPS when there is only a single slot"""