Minor change in controller