fixes the failing test