diff --git a/checkm8.py b/checkm8.py
index 67ab1e06..900c6635 100644
--- a/checkm8.py
+++ b/checkm8.py
@@ -683,7 +683,7 @@ def exploit(match=None):
     print('(%0.2f seconds)' % (time.time() - start))
     dfu.release_device(device)
 
-def exploit_a8_a9():
+def exploit_a8_a9(match=None):
   print('*** checkm8 exploit by axi0mX ***')
 
   device = dfu.acquire_device(match=match)
diff --git a/ipwndfu b/ipwndfu
index ca1c3646..799a5a38 100755
--- a/ipwndfu
+++ b/ipwndfu
@@ -169,7 +169,7 @@ def pwn(device=None, match_device=None):
     elif serial.cpid in ['8947', '8950', '8960', '8002', '8004', '8010', '8011', '8012', '8015']:
         checkm8.exploit(match=match_device)
     elif serial.cpid in ['7000', '8000', '8003']:
-        checkm8.exploit_a8_a9()
+        checkm8.exploit_a8_a9(match=match_device)
     else:
         print('Found: ' + serial_number, file=sys.stderr)
         print('ERROR: This device is not supported.', file=sys.stderr)