fixed the initial failing test