minor fix to controller