fix the failing test