Update unit test to work with the own controllers.