TST: improve last test