Updated Controller unit test.