diff --git a/tests/cmd/test_query.rs b/tests/cmd/test_query.rs
index 3274ca0..b67c752 100644
--- a/tests/cmd/test_query.rs
+++ b/tests/cmd/test_query.rs
@@ -82,7 +82,7 @@ fn test_query_save_png() {
         test_query,
         "--save",
         "--save-dir",
-        "./",
+        save_dir,
         "--file-type",
         file_type,
         "--image-column",