fix in unit test