Fix travis ci

I found master branch fails travis ci test now. I tried to fix it.