TST: fix test errors on travis-ci