Made failing test pass