Fix up failing test