Fix it up for travis