diff --git a/examples/run_expt.py b/examples/run_expt.py
index 81a11708d5..7de121be29 100644
--- a/examples/run_expt.py
+++ b/examples/run_expt.py
@@ -3,7 +3,7 @@
 
 from sklearn.metrics import accuracy_score
 
-from tableshift.core import get_dataset
+from tableshift import get_dataset
 from tableshift.models.training import train
 from tableshift.models.utils import get_estimator