Fixed get_driver_module to support embedded mode