diff --git a/src/Interface/BadgeableInterface.php b/src/Interface/BadgeableInterface.php
index 29a23e8..ad71df5 100644
--- a/src/Interface/BadgeableInterface.php
+++ b/src/Interface/BadgeableInterface.php
@@ -4,7 +4,7 @@ declare(strict_types=1);
 
 namespace Pcm\BadgeBundle\Interface;
 
-use Pcm\BadgeBundle\Enum\Badge;
+use Pcm\BadgeBundle\Enum\BadgeColour;
 
 /**
  * Allows rendering the implementing class as a badge using the PCM Badge twig component.
@@ -14,8 +14,8 @@ interface BadgeableInterface
     /**
      * Get the badge colour that should be used when rendering this object as a badge.
      *
-     * @return Badge Colour of badge to render
+     * @return BadgeColour Colour of badge to render
      */
-    public function getBadgeColour(): Badge;
+    public function getBadgeColour(): BadgeColour;
 }
 
diff --git a/src/Twig/Component/Badge.php b/src/Twig/Component/Badge.php
index e6df824..d6a208c 100644
--- a/src/Twig/Component/Badge.php
+++ b/src/Twig/Component/Badge.php
@@ -4,7 +4,7 @@ declare(strict_types=1);
 
 namespace Pcm\BadgeBundle\Twig\Component;
 
-use Pcm\BadgeBundle\Enum\Badge as EnumBadge;
+use Pcm\BadgeBundle\Enum\BadgeColour;
 use Pcm\BadgeBundle\Interface\BadgeableInterface;
 use Symfony\UX\TwigComponent\Attribute\AsTwigComponent;
 use TailwindMerge\TailwindMerge;
@@ -44,7 +44,7 @@ final class Badge
         }
 
         if ($colour) {
-            $cases = array_map(fn (EnumBadge $b) => strtolower($b->name), EnumBadge::cases());
+            $cases = array_map(fn (BadgeColour $b) => strtolower($b->name), BadgeColour::cases());
 
             if (!in_array($colour, $cases)) {
                 $formattedCases = implode(', ', array_map(fn (string $s) => '"'.$s.'"', $cases));
@@ -52,7 +52,7 @@ final class Badge
             }
 
             $colour  = strtoupper($colour);
-            $palette = EnumBadge::{$colour}->getPalette();
+            $palette = BadgeColour::{$colour}->getPalette();
         }
 
         $merger = TailwindMerge::instance();