Added a fix for Travis-CI setup