Fixing up test script for Travis