Added fix for Travis CI