Fix test stubs