update unit tests from previous change