diff --git a/tests/extmod/machine1.py b/tests/extmod/machine1.py
index af4dacd03a..433a180376 100644
--- a/tests/extmod/machine1.py
+++ b/tests/extmod/machine1.py
@@ -1,7 +1,10 @@
 # test machine module
 
 try:
-    import machine
+    try:
+        import umachine as machine
+    except ImportError:
+        import machine
 except ImportError:
     print("SKIP")
     import sys