Fix travis-ci test issue