Fix for travis-ci