Fixing build

Fixing Travis builds.