Fix test failure after entity_id/parent_id change.