diff --git a/.github/workflows/_basic-test.yml b/.github/workflows/_basic-test.yml
index 8f64c12e..a70d5897 100644
--- a/.github/workflows/_basic-test.yml
+++ b/.github/workflows/_basic-test.yml
@@ -2,7 +2,7 @@ name: test-basics
 on: [workflow_call, workflow_dispatch]
 
 jobs:
-  Ubuntu:
+  cpu:
     runs-on: ubuntu-latest #ToDo: ${{ needs.build.outputs.os }}
     strategy:
       fail-fast: false
diff --git a/.github/workflows/_demo-test.yml b/.github/workflows/_demo-test.yml
index 02048892..d23002d5 100644
--- a/.github/workflows/_demo-test.yml
+++ b/.github/workflows/_demo-test.yml
@@ -2,7 +2,7 @@ name: test-examples
 on: [workflow_call, workflow_dispatch]
 
 jobs:
-  GCP:
+  gpu:
     runs-on: self-hosted #ToDo: ${{ needs.build.outputs.os }}
     strategy:
       fail-fast: false
diff --git a/tests/notebook_testing.py b/tests/notebook_testing.py
index 6419e6a2..827743c8 100644
--- a/tests/notebook_testing.py
+++ b/tests/notebook_testing.py
@@ -170,8 +170,8 @@ def startTest(self, test):
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
-    parser.add_argument("-notebook_path", help="Path to the notebook file")
-    parser.add_argument("-module", help="Can either test examples or Basics")
+    parser.add_argument("notebook_path", help="Path to the notebook file")
+    parser.add_argument("module", help="Can either test examples or Basics")
     args = parser.parse_args()
 
     NotebookTest.notebook = args.notebook_path
diff --git a/tests/test_demos.sh b/tests/test_demos.sh
index 7e35ecfc..4b8b1ea1 100755
--- a/tests/test_demos.sh
+++ b/tests/test_demos.sh
@@ -14,4 +14,5 @@ echo -n "$1" > .ivy/key.pem
 python3 -m pip install -r requirements.txt >/dev/null 2>&1
 
 # run test
-python3 tests/notebook_testing.py -notebook-path "$2" -module "$3"
\ No newline at end of file
+echo "$2" "$3"
+python3 tests/notebook_testing.py "$2" "$3"
\ No newline at end of file