small bug fix in Controller