Fix build

Fix build on CI travis.