diff --git a/tests/extmod/machine1.py b/tests/extmod/machine1.py
index af4dacd03a01f362153288e2034d82566fdee6c0..433a180376cdf2af49c6c6cfbd3cdda930b0bb57 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