TST : fix failing test