diff --git a/lua/Polydev/python.lua b/lua/Polydev/python.lua
index c6ecd52..e515d1a 100644
--- a/lua/Polydev/python.lua
+++ b/lua/Polydev/python.lua
@@ -222,7 +222,7 @@ function M.custom_run()
 	    venv_cmd = vim.fn.filereadable(venv_bash) == 1 and "source " .. venv_bash .. " && " or ""
 	end
 
-	M.open_float_terminal(venv_cmd .. "python3 " .. custom_file)
+	M.open_float_terminal(venv_cmd .. "python3 " .. custom_file .. ".py")
     end)
 end