Add a test for controllers