Fix Category controller