Fix unit test

come on travis