Update unit tests.
(cherry picked from commit 151b1b818378c9e91313974c3f1050bd415b8de9)