diff --git a/imperative/src/version.ld b/imperative/src/version.ld
index 19c7cde5..876e7ddb 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 733a8cee..d380d389 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
         )