Fixing travis-ci build