Fixed for Travis CI test