Fix the test stubs