Revert this change back.

Fix test.