fix deprecation warning: replace module_name with model_name