Fixing mistake in controller.