Fixing Controller unit test.