Fix test.

Fix test.