Fix unit test.

Fix unit test after some core data changes.