Fix travis tests. Closes #2