diff --git a/ast/asdl_rs.py b/ast/asdl_rs.py
index b217e629a7..8caeca9014 100755
--- a/ast/asdl_rs.py
+++ b/ast/asdl_rs.py
@@ -474,7 +474,7 @@ class ExtendModuleVisitor(EmitVisitor):
     def visitModule(self, mod):
         depth = 0
         self.emit(
-            "pub fn extend_module_nodes(vm: &VirtualMachine, module: &PyObject) {",
+            "pub fn extend_module_nodes(vm: &VirtualMachine, module: &Py<PyModule>) {",
             depth,
         )
         self.emit("extend_module!(vm, module, {", depth + 1)