Fix test for travis-ci