diff --git a/test/run.py b/test/run.py
index 7df33ca1..b7ce5938 100755
--- a/test/run.py
+++ b/test/run.py
@@ -304,8 +304,7 @@ if __name__ == '__main__':
     import sys
 
     if arguments['--debug']:
-        from jedi import api, debug
-        api.set_debug_function(debug.print_to_stdout)
+        jedi.set_debug_function()
 
     # get test list, that should be executed
     test_files = {}
diff --git a/test/test_regression.py b/test/test_regression.py
index 64102d0f..6081e9b5 100644
--- a/test/test_regression.py
+++ b/test/test_regression.py
@@ -12,7 +12,7 @@ import jedi
 from jedi import Script
 from jedi import api, parsing
 
-#jedi.set_debug_function(jedi.debug.print_to_stdout)
+#jedi.set_debug_function()
 
 
 class TestRegression(TestCase):