1 files changed, 9 insertions, 3 deletions
diff --git a/tests/extmod/machine_pinbase.py b/tests/extmod/machine_pinbase.py
index 07a489a596..5e82823ec0 100644
--- a/tests/extmod/machine_pinbase.py
+++ b/tests/extmod/machine_pinbase.py
@@ -1,10 +1,16 @@
 try:
-    from umachine import PinBase
+    import umachine as machine
 except ImportError:
-    from machine import PinBase
+    import machine
+try:
+    machine.PinBase
+except AttributeError:
+    print("SKIP")
+    import sys
+    sys.exit()
 
 
-class MyPin(PinBase):
+class MyPin(machine.PinBase):
 
     def __init__(self):
         print("__init__")