diff --git a/tests/yen_test.py b/tests/yen_test.py
index e253fb4..fb36090 100644
--- a/tests/yen_test.py
+++ b/tests/yen_test.py
@@ -188,7 +188,7 @@ def test_yen_install_module(yen_path: str) -> None:
 @parametrize_python_and_rust_path
 def test_yen_run(yen_path: str) -> None:
     output = run([yen_path, "run", "morsedecode", "--", "....", "..", "-.-.--"])
-    # assert output == "HI!\n"
+    assert output == "HI!\n"
 
     run([yen_path, "install", "astmath", "--module", "astmath"])
     output = run([yen_path, "run", "astmath", "3 * 3"])
diff --git a/yen-rs/src/commands/run.rs b/yen-rs/src/commands/run.rs
index 415d4c4..c276aaf 100644
--- a/yen-rs/src/commands/run.rs
+++ b/yen-rs/src/commands/run.rs
@@ -30,9 +30,9 @@ pub async fn execute(args: Args) -> miette::Result<()> {
     let (_, python_bin_path) = ensure_python(args.python).await?;
 
     let package_name = args.package_name;
-    install_package(&package_name, python_bin_path, &package_name, false, false).await?;
+    let (shim_path, _) =
+        install_package(&package_name, python_bin_path, &package_name, false, false).await?;
 
-    let shim_path = PACKAGE_INSTALLS_PATH.join(package_name);
     let output = Command::new(shim_path)
         .args(args.run_args)
         .stdin(Stdio::inherit())