minor tweek to controller