need to fix test for controller