Correct arbitrary suffix

johentsch · Aug 11, 2023 · b5a4019 · b5a4019
1 parent b34b221
commit b5a4019
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/ms3/bs4_parser.py b/src/ms3/bs4_parser.py
@@ -2298,8 +2298,8 @@ def set_instrument(self, staff_id: Union[str, int], trackname):
         trackname_norm = trackname.lower().strip('.')
         if trackname_norm not in self.key2default_instrumentation:
             # add splitting by suffix and then adapt other names to it
-            split_trackname = trackname_norm.split()
-            trackname_without_suffix = " ".join(split_trackname[:-1])
+            split_trackname = trackname.split()
+            trackname_without_suffix = " ".join(split_trackname[:-1]).lower().strip('.')
             if trackname_without_suffix in self.key2default_instrumentation:
                 suffix = split_trackname[-1]
                 new_values = self.add_suffix(self.key2default_instrumentation[trackname_without_suffix], suffix)