Fix undefined controller error.