Fix for Travis test script