Fix plimc error when use custom_module