diff --git a/expo/experimenter/autosklearn.py b/expo/experimenter/autosklearn.py
index 9d0ea2df4..02a3cc465 100644
--- a/expo/experimenter/autosklearn.py
+++ b/expo/experimenter/autosklearn.py
@@ -24,7 +24,8 @@ class ASRunner:
         )
 
     def run(self):
-        import autosklearn
+        import autosklearn.classification
+        import autosklearn.regression
 
         train_path = self.datasets["train"]
         dev_wo_target_path = self.datasets["dev_wo_target"]