Fix test on travis-ci