add test for the controller