diff --git a/artiq/tools.py b/artiq/tools.py
index d98059356..1c4949a23 100644
--- a/artiq/tools.py
+++ b/artiq/tools.py
@@ -1,5 +1,6 @@
 import asyncio
-import importlib
+import importlib.util
+import importlib.machinery
 import inspect
 import logging
 import os