fix the misleading model_key name