fix update unit tests