diff --git a/dolo/tests/test_customdr.py b/dolo/tests/test_customdr.py
index <HASH>..<HASH> 100644
--- a/dolo/tests/test_customdr.py
+++ b/dolo/tests/test_customdr.py
@@ -1,6 +1,6 @@
 def test_custom_dr():
 
-    from dolo import yaml_import
+    from dolo import yaml_import, simulate, time_iteration
     import numpy as np
     from dolo.numeric.decision_rule import CustomDR
FIX: missing imports in tests.