fix Unit test