diff --git a/imperative/src/version.ld b/imperative/src/version.ld
index 19c7cde52ae01019004a1e076c5ee4b387da7892..876e7ddb6103e26c2b79f24a926f33f9b5dcb373 100644
--- a/imperative/src/version.ld
+++ b/imperative/src/version.ld
@@ -4,6 +4,7 @@ global:
     MEGDNN_VSYM_*;
     mgb_get_extern_c_opr_api_versioned;
     PyInit__imperative_rt;
+    PyInit__mgb;
     extern "C++" {
         *mgb::*;
         *megdnn::*;
diff --git a/src/CMakeLists.txt b/src/CMakeLists.txt
index 733a8cee325fe284c837d892e1938f7eae8fd54c..d380d389eab3940b316f470e83ecad6068b25215 100644
--- a/src/CMakeLists.txt
+++ b/src/CMakeLists.txt
@@ -181,11 +181,7 @@ if(ANDROID)
     target_link_libraries(megbrain PUBLIC log)
 endif()
 
-if(MGE_BUILD_IMPERATIVE_RT)
-    set (_VER_FILE ${PROJECT_SOURCE_DIR}/python_module/src/version.ld)
-else()
-    set (_VER_FILE ${PROJECT_SOURCE_DIR}/imperative/src/version.ld)
-endif()
+set (_VER_FILE ${PROJECT_SOURCE_DIR}/imperative/src/version.ld)
 
 if(MGE_BUILD_IMPERATIVE_RT
         )