Correct controller error
Correct controller error