diff --git a/tests/goodixmoc/custom.py b/tests/goodixmoc/custom.py
old mode 100755
new mode 100644
index 38fdd26c..73f90a53
--- a/tests/goodixmoc/custom.py
+++ b/tests/goodixmoc/custom.py
@@ -31,7 +31,7 @@ d.clear_storage_sync()
 template = FPrint.Print.new(d)
 
 def enroll_progress(*args):
-    assert d.get_finger_status() == FPrint.FingerStatusFlags.NEEDED
+    assert d.get_finger_status() & FPrint.FingerStatusFlags.NEEDED
     print('enroll progress: ' + str(args))
 
 def identify_done(dev, res):
diff --git a/tests/synaptics/custom.py b/tests/synaptics/custom.py
index 0a88ef80..5b4b1481 100755
--- a/tests/synaptics/custom.py
+++ b/tests/synaptics/custom.py
@@ -51,7 +51,7 @@ d.clear_storage_sync()
 template = FPrint.Print.new(d)
 
 def enroll_progress(*args):
-    #assert d.get_finger_status() == FPrint.FingerStatusFlags.NEEDED
+    assert d.get_finger_status() & FPrint.FingerStatusFlags.NEEDED
     print('enroll progress: ' + str(args))
 
 # List, enroll, list, verify, delete, list