diff --git a/kernels/convolution_milo.py b/kernels/convolution_milo.py
index de80db5..6a1c3a7 100644
--- a/kernels/convolution_milo.py
+++ b/kernels/convolution_milo.py
@@ -140,14 +140,14 @@ def tune(
 
 
 if __name__ == "__main__":
-    language = sys.argv[1]
-    device_name = sys.argv[2]
-
     if len(sys.argv) != 3:
         raise ValueError(
-            f"Usage: python convolution_milo.py [language ('HIP' or 'CUDA')] [device name], given: {sys.argv}"
+            f"Usage: python convolution_milo.py [language ('HIP' or 'CUDA')] [device name], given: {sys.argv[1:]}"
         )
 
+    language = sys.argv[1]
+    device_name = sys.argv[2]
+
     if language not in ("HIP", "CUDA"):
         raise ValueError(f"{language} not valid, specify HIP or CUDA")
 
diff --git a/kernels/dedisp_milo/dedispersion_milo.py b/kernels/dedisp_milo/dedispersion_milo.py
index 15e07f1..ec74f76 100755
--- a/kernels/dedisp_milo/dedispersion_milo.py
+++ b/kernels/dedisp_milo/dedispersion_milo.py
@@ -140,14 +140,14 @@ def tune(
 
 
 if __name__ == "__main__":
-    language = sys.argv[1]
-    device_name = sys.argv[2]
-
     if len(sys.argv) != 3:
         raise ValueError(
-            f"Usage: python dedispersion_milo.py [language ('HIP' or 'CUDA')] [device name], given: {sys.argv}"
+            f"Usage: python dedispersion_milo.py [language ('HIP' or 'CUDA')] [device name], given: {sys.argv[1:]}"
         )
 
+    language = sys.argv[1]
+    device_name = sys.argv[2]
+
     if language not in ("HIP", "CUDA"):
         raise ValueError(f"{language} not valid, specify HIP or CUDA")
 
diff --git a/kernels/gemm_milo.py b/kernels/gemm_milo.py
index 3d8a0d3..f0c7dbe 100644
--- a/kernels/gemm_milo.py
+++ b/kernels/gemm_milo.py
@@ -177,12 +177,12 @@ def tune(
 
 
 if __name__ == "__main__":
+    if len(sys.argv) != 3:
+        raise ValueError(f"Usage: python gemm_milo.py [language ('HIP', 'OpenCL' or 'CUDA')] [device name], given: {sys.argv[1:]}")
+
     language = sys.argv[1]
     device_name = sys.argv[2]
 
-    if len(sys.argv) != 3:
-        raise ValueError(f"Usage: python gemm_milo.py [language ('HIP', 'OpenCL' or 'CUDA')] [device name], given: {sys.argv}")
-
     if language not in ("HIP", "OpenCL", "CUDA"):
         raise ValueError(f"{language} not valid, specify HIP, OpenCL or CUDA")
 
diff --git a/kernels/hotspot_milo.py b/kernels/hotspot_milo.py
index c383790..80d9c91 100644
--- a/kernels/hotspot_milo.py
+++ b/kernels/hotspot_milo.py
@@ -254,12 +254,12 @@ def get_results(self):
 
 
 if __name__ == "__main__":
+    if len(sys.argv) != 3:
+        raise ValueError(f"Usage: python hotspot_milo.py [language ('HIP' or 'CUDA')] [device name], given: {sys.argv[1:]}")
+
     language = sys.argv[1]
     device_name = sys.argv[2]
 
-    if len(sys.argv) != 3:
-        raise ValueError(f"Usage: python hotspot_milo.py [language ('HIP' or 'CUDA')] [device name], given: {sys.argv}")
-
     if language not in ("HIP", "CUDA"):
         raise ValueError(f"{language} not valid, specify HIP or CUDA")