TST: fixing the failing case