Fix the failing test