Fix controller's test