Fix Travis CI
Merge pull request #5 from jedmao/fix-travis

Fix Travis CI