add controller after unit test