Fix unit test.

Change-Id: I45845635ead25a4b0cef454080e1d8f2a28b1c6e