Fix controller error