Fix test script for travis