fixed unit tests for controller to get working again