Fixing the build. Hope it works on Travis