trying to fix the failing test