Fix typo in tests.

Fix typo in tests.