diff --git a/derive/src/pymodule.rs b/derive/src/pymodule.rs
index b861f9565..97959b8ba 100644
--- a/derive/src/pymodule.rs
+++ b/derive/src/pymodule.rs
@@ -63,7 +63,7 @@ pub fn impl_pymodule(
         },
         parse_quote! {
             pub(crate) fn extend_module(
-                vm: &::rustpython_vm::vm::VirtualMachine,
+                vm: &::rustpython_vm::VirtualMachine,
                 module: &::rustpython_vm::pyobject::PyObjectRef,
             ) {
                 #module_extend_items
@@ -72,7 +72,7 @@ pub fn impl_pymodule(
         parse_quote! {
             #[allow(dead_code)]
             pub(crate) fn make_module(
-                vm: &::rustpython_vm::vm::VirtualMachine
+                vm: &::rustpython_vm::VirtualMachine
             ) -> ::rustpython_vm::pyobject::PyObjectRef {
                 let module = vm.new_module(MODULE_NAME, vm.ctx.new_dict());
                 extend_module(vm, &module);