Fix model name in controller