Fixing TravisCI tests