fix Unit Test