Fixed failing test stubs :P