Fix unit test
1.0.1