Fix travis build failure

Travis build failed and local passed. This fixes travis build.