Fixed travis-ci failing unit test.