Fix the CI tests

Tested. Fixed the CI tests on travis-ci