unit tests update