Fix to work with travis-ci