Fix travis build

Not sure this is the way to go though.