diff --git a/google/colab/_shell.py b/google/colab/_shell.py
index 7bf3f427..91acab10 100644
--- a/google/colab/_shell.py
+++ b/google/colab/_shell.py
@@ -89,8 +89,8 @@ def get_code_name(self, raw_code, code, number):
     """
     code_name = super().get_code_name(raw_code, code, number)
     if code_name.endswith('.py'):
-      name = pathlib.Path(code_name).with_suffix('').name
-      code_name = f'<ipython-input-{number}-{name}>'
+      path = pathlib.Path(code_name)
+      code_name = f'ipython-input-{number}-{path.name}'
     return code_name