diff --git a/src/tests.rs b/src/tests.rs
index 99fa30d..afd6cb6 100644
--- a/src/tests.rs
+++ b/src/tests.rs
@@ -45,14 +45,14 @@ mod tests {
         let env = format!("KEEP_DIR={} cargo run -- --verbose", dir.display());
 
         // Create first item with tag_a and tag
-        let cmd = format!("echo {} | {} tag tag_a tag", input_a, env);
+        let cmd = format!("echo {} | {} tag tag_a tag", INPUT_A, env);
         let output = run_sh(cmd.as_str());
-        assert!(output.status.success(), "Command failed: {}", input_a);
+        assert!(output.status.success(), "Command failed: {}", INPUT_A);
 
         // Create second item with tag_b and tag
-        let cmd = format!("echo {} | {} tag tag_b tag", input_b, env);
+        let cmd = format!("echo {} | {} tag tag_b tag", INPUT_B, env);
         let output = run_sh(cmd.as_str());
-        assert!(output.status.success(), "Command failed: {}", input_b);
+        assert!(output.status.success(), "Command failed: {}", INPUT_B);
     }
 
     // Helper function to check if a file exists in the data directory