diff --git a/src/main/kotlin/net/theevilreaper/stelaris/cli/generator/dart/bossbar/BossBarFlagGenerator.kt b/src/main/kotlin/net/theevilreaper/stelaris/cli/generator/dart/bossbar/BossBarFlagGenerator.kt
index a68523a..42e4b98 100644
--- a/src/main/kotlin/net/theevilreaper/stelaris/cli/generator/dart/bossbar/BossBarFlagGenerator.kt
+++ b/src/main/kotlin/net/theevilreaper/stelaris/cli/generator/dart/bossbar/BossBarFlagGenerator.kt
@@ -24,19 +24,19 @@ class BossBarFlagGenerator : BaseGenerator(
             .type(
                 ClassSpec.enumClass(className)
                     .apply {
-                        BossBar.Overlay.entries.forEach { overlay ->
+                        BossBar.Flag.entries.forEach { flag ->
                             enumProperty(
-                                EnumEntrySpec.builder(overlay.name.lowercase())
+                                EnumEntrySpec.builder(flag.name.lowercase())
                                     .parameter(
                                         EnumParameterSpec.positional(
                                             "%C",
-                                            StringHelper.mapDisplayName(overlay.name)
+                                            StringHelper.mapDisplayName(flag.name)
                                         )
                                     )
                                     .parameter(
                                         EnumParameterSpec.positional(
                                             "%C",
-                                            overlay.name.uppercase()
+                                            flag.name.uppercase()
                                         )
                                     )
                                     .build()
diff --git a/src/test/kotlin/net/theevilreaper/stelaris/cli/generator/dart/bossbar/BossBarFlagGeneratorTest.kt b/src/test/kotlin/net/theevilreaper/stelaris/cli/generator/dart/bossbar/BossBarFlagGeneratorTest.kt
index 07d4309..244acc9 100644
--- a/src/test/kotlin/net/theevilreaper/stelaris/cli/generator/dart/bossbar/BossBarFlagGeneratorTest.kt
+++ b/src/test/kotlin/net/theevilreaper/stelaris/cli/generator/dart/bossbar/BossBarFlagGeneratorTest.kt
@@ -33,11 +33,9 @@ class BossBarFlagGeneratorTest : GenerationTestBase() {
         val expectedClass = """
             enum BossBarFlag {
 
-              progress('Progress', 'PROGRESS'),
-              notched_6('Notched 6', 'NOTCHED_6'),
-              notched_10('Notched 10', 'NOTCHED_10'),
-              notched_12('Notched 12', 'NOTCHED_12'),
-              notched_20('Notched 20', 'NOTCHED_20');
+              darken_screen('Darken Screen', 'DARKEN_SCREEN'),
+              play_boss_music('Play Boss Music', 'PLAY_BOSS_MUSIC'),
+              create_world_fog('Create World Fog', 'CREATE_WORLD_FOG');
 
               final String displayName;
               final String type;