Fix failing test stubs