Fixing build on travis-ci