Fix controller spec