minor fix in the controller