Fixing travis-ci test