diff --git a/google/colab/_shell.py b/google/colab/_shell.py
index fcbe6cbe..63ea6f1f 100644
--- a/google/colab/_shell.py
+++ b/google/colab/_shell.py
@@ -90,7 +90,7 @@ def get_code_name(self, raw_code, code, number):
     code_name = super().get_code_name(raw_code, code, number)
     if code_name.endswith('.py'):
       path = pathlib.Path(code_name)
-      code_name = f'/tmp/ipython-input-{number}-{path.name}'
+      code_name = f'{path.parent}/ipython-input-{number}-{path.name}'
     return code_name