Minor change to controller