From 96a5bb7c4b0db5c46a73eff5d867b4279a3ec40f Mon Sep 17 00:00:00 2001
From: jakob <j.wasserthal@gmx.de>
Date: Fri, 17 Jan 2025 10:14:11 +0100
Subject: [PATCH] bugfix

---
 totalsegmentator/bin/totalseg_download_weights.py | 4 ++--
 totalsegmentator/map_to_binary.py                 | 3 ++-
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/totalsegmentator/bin/totalseg_download_weights.py b/totalsegmentator/bin/totalseg_download_weights.py
index 430c0f52d..da8d7872b 100644
--- a/totalsegmentator/bin/totalseg_download_weights.py
+++ b/totalsegmentator/bin/totalseg_download_weights.py
@@ -41,7 +41,6 @@ def main():
         "lung_vessels": [258],
         "cerebral_bleed": [150],
         "hip_implant": [260],
-        "coronary_arteries": [507],
         "pleural_pericard_effusion": [315],
         "body": [299],
         "body_fast": [300],
@@ -69,7 +68,8 @@ def main():
         "face_mr": [856],
         "brain_structures": [409],
         "thigh_shoulder_muscles": [857],
-        "thigh_shoulder_muscles_mr": [857]
+        "thigh_shoulder_muscles_mr": [857],
+        "coronary_arteries": [507]
     }
 
     setup_totalseg()
diff --git a/totalsegmentator/map_to_binary.py b/totalsegmentator/map_to_binary.py
index 8c2afd468..0d26811d7 100644
--- a/totalsegmentator/map_to_binary.py
+++ b/totalsegmentator/map_to_binary.py
@@ -603,7 +603,8 @@
     "face_mr": 856,
     "brain_structures": 409,
     "thigh_shoulder_muscles": 857,
-    "thigh_shoulder_muscles_mr": 857
+    "thigh_shoulder_muscles_mr": 857,
+    "coronary_arteries": 507
 }