Fixed controller test (now all tests will pass)