Fix for 0.6 testing on travis