TST: fixed failing test in test_with_na 2