diff --git a/tests/samples/tsp_script.py b/tests/samples/tsp_script.tsp
similarity index 98%
rename from tests/samples/tsp_script.py
rename to tests/samples/tsp_script.tsp
index 8107bd15..60da9791 100644
--- a/tests/samples/tsp_script.py
+++ b/tests/samples/tsp_script.tsp
@@ -1,3 +1,2 @@
 """Sample script used in test_smu.py."""
-
 print("TEK")
diff --git a/tests/test_smu.py b/tests/test_smu.py
index 4345be73..679a8809 100644
--- a/tests/test_smu.py
+++ b/tests/test_smu.py
@@ -49,7 +49,7 @@ def test_smu(  # noqa: PLR0915
         assert "Query" in stdout
 
     smu.load_script(
-        file_path=f"{Path(os.path.realpath(__file__)).parent}/samples/tsp_script.py",
+        file_path=f"{Path(os.path.realpath(__file__)).parent}/samples/tsp_script.tsp",
         run_script=True,
         to_nv_memory=True,
     )
@@ -61,7 +61,7 @@ def test_smu(  # noqa: PLR0915
     assert "loadfuncs()" in stdout
     smu.expect_esr(0)
     smu.load_script(
-        file_path=f"{Path(os.path.realpath(__file__)).parent}/samples/tsp_script.py",
+        file_path=f"{Path(os.path.realpath(__file__)).parent}/samples/tsp_script.tsp",
         script_name="tsp_function",
     )
     stdout = capsys.readouterr().out