Fix controller to get test to run