Minor change to  controller