fix unit test for travis-ci.