update unit test codes