diff --git a/tests/CMakeLists.txt b/tests/CMakeLists.txt
index b28fc402..c3b0fa05 100644
--- a/tests/CMakeLists.txt
+++ b/tests/CMakeLists.txt
@@ -29,7 +29,7 @@ ADDTEST(ams_inference_double torch_model.cpp AMSInferDouble ${CMAKE_CURRENT_SOUR
 ADDTEST(ams_inference_single torch_model.cpp AMSInferSingle ${CMAKE_CURRENT_SOURCE_DIR}/debug_model.pt "single")
 endif()
 
-add_test(NAME AMSInferSingleDeltaUQ::HOST COMMAND  ams_example --precision single --uqtype deltauq-mean -db ./db -S /p/vast1/ggeorgak/projects/ams/deltauq/tuple-single.torchscript -e 100)
+add_test(NAME AMSInferSingleDeltaUQ::HOST COMMAND  ams_example --precision single --uqtype deltauq-mean -db ./db -S ${CMAKE_CURRENT_SOURCE_DIR}/tuple-single.torchscript -e 100)
 if(WITH_FAISS)
 ADDTEST(ams_hdcache_mean_double test_hdcache.cpp AMSHDCacheMeanPolicyDouble ${CMAKE_CURRENT_SOURCE_DIR}/faiss_debug.pt "double" 1 10 4.0 4 5)
 # The max case fails on DEVICE. We should be aware abou this when adding support for CI for GPUs
diff --git a/tests/tuple-single.torchscript b/tests/tuple-single.torchscript
new file mode 100644
index 00000000..c40411b4
Binary files /dev/null and b/tests/tuple-single.torchscript differ