diff --git a/bin/i3-companion b/bin/i3-companion
index 91d7fc7..2fff319 100755
--- a/bin/i3-companion
+++ b/bin/i3-companion
@@ -357,7 +357,12 @@ async def workspace_rename(i3, event):
             cls = (window.window_class or "").lower()
             if cls in application_icons_ignore:
                 continue
-            icon = application_icons.get(cls, application_icons.get(cls.split("-")[0]))
+            icon = application_icons.get(
+                cls,
+                application_icons.get(
+                    cls.split("-")[0], application_icons.get(cls.split("_")[0])
+                ),
+            )
             icons.add(icon or application_icons_nomatch)
         if any([i not in application_icons_alone for i in icons]):
             icons -= application_icons_alone