Fix script for travis