correct code on failing test