Modification for Travis CI