Fix tests on Travis
Merge pull request #250 from janko-m/fix-tests

Fix tests on Travis