temporary fix to get user_model_trainer working.