diff --git a/src/finn/core/remote_exec.py b/src/finn/core/remote_exec.py
index e97eb19a101e83f9d9603637e131b2ec9b7d16a4..afea8f74fdcf416b5b7d634f2c63406f6b6efe47 100644
--- a/src/finn/core/remote_exec.py
+++ b/src/finn/core/remote_exec.py
@@ -65,7 +65,7 @@ def remote_exec(model, execution_context):
     cmd = (
         "sshpass -p {} ssh {}@{} -p {} "
         '"cd {}/{}; echo "{}" | '
-        'sudo -S python3.6 driver.py remote_pynq 1 resizer.bit input.npy output.npy"'
+        'sudo -S python3.6 driver.py --exec_mode="remote_pynq" --batchsize=1"'
     ).format(
         pynq_password,
         pynq_username,
diff --git a/src/finn/core/throughput_test.py b/src/finn/core/throughput_test.py
index fc929237bf6c985997e49cc3f74c7d492d79839a..c82d540e29fc59b92a22bf011e823a9f8c076843 100644
--- a/src/finn/core/throughput_test.py
+++ b/src/finn/core/throughput_test.py
@@ -47,8 +47,7 @@ def throughput_test(model):
     cmd = (
         "sshpass -p {} ssh {}@{} -p {} "
         '"cd {}/{}; echo "{}" | '
-        "sudo -S python3.6 driver.py throughput_test 1000 "
-        'resizer.bit input.npy output.npy"'
+        'sudo -S python3.6 driver.py --exec_mode="throughput_test" --batchsize=1000"'
     ).format(
         pynq_password,
         pynq_username,