fixed the failing test