Fixed the tests

The Mojo upgrade broke the Testcase.