Fix travis build

Closes #5