Fixed 1 line in controller