diff --git a/backend/pashub_fetcher/pashub_client.py b/backend/pashub_fetcher/pashub_client.py index 20b8590d..11195960 100644 --- a/backend/pashub_fetcher/pashub_client.py +++ b/backend/pashub_fetcher/pashub_client.py @@ -87,6 +87,9 @@ class PashubClient: return None def _get_core_file_type(self, file: EvidenceFileData) -> Optional[CoreFiles]: + if file.evidence_category == "retrofit design": + return CoreFiles.RETROFIT_DESIGN_DOC + for core_file in CoreFiles: if file.file_name.startswith(core_file.value): return core_file