Fix Travis tests

This fixes a bug introduced in #180.
Merge pull request #181 from myint/fix

Fix Travis tests