diff --git a/src/linalg.h b/src/linalg.h
index 543c11ca..802ec484 100644
--- a/src/linalg.h
+++ b/src/linalg.h
@@ -694,4 +694,11 @@ inline PyObject* py_var(VM* vm, const PyVec3& obj){ return VAR_T(PyVec3, obj);}
 inline PyObject* py_var(VM* vm, const Mat3x3& obj){ return VAR_T(PyMat3x3, obj); }
 inline PyObject* py_var(VM* vm, const PyMat3x3& obj){ return VAR_T(PyMat3x3, obj); }
 
+inline void add_module_linalg(VM* vm){
+    PyObject* linalg = vm->new_module("linalg");
+    PyVec2::register_class(vm, linalg);
+    PyVec3::register_class(vm, linalg);
+    PyMat3x3::register_class(vm, linalg);
+}
+
 }   // namespace pkpy
\ No newline at end of file
diff --git a/src/pocketpy.h b/src/pocketpy.h
index 870a85b2..36c89a06 100644
--- a/src/pocketpy.h
+++ b/src/pocketpy.h
@@ -1053,9 +1053,7 @@ inline void VM::post_init(){
         add_module_requests(this);
     }
 
-    PyObject* linalg = new_module("linalg");
-    PyVec2::register_class(this, linalg);
-    PyMat3x3::register_class(this, linalg);
+    add_module_linalg(this);
 #endif
 }