diff --git a/import2ledger/hooks/add_entity.py b/import2ledger/hooks/add_entity.py index ccf6ad9..90321b3 100644 --- a/import2ledger/hooks/add_entity.py +++ b/import2ledger/hooks/add_entity.py @@ -68,3 +68,5 @@ class AddEntityHook: def run(self, data): if ('payee' in data) and ('entity' not in data): data['entity'] = self._name2entity(data['payee'], 1) + if ('corporation' in data) and ('corp_entity' not in data): + data['corp_entity'] = self._name2entity(data['corporation'], 0)