Travis fix

Add fix for travis_ci error