diff --git a/DependencyInjection/EnumExtension.php b/DependencyInjection/EnumExtension.php
index 75944f8..6887da4 100644
--- a/DependencyInjection/EnumExtension.php
+++ b/DependencyInjection/EnumExtension.php
@@ -10,7 +10,7 @@
 /**
  * @author Yann Eugoné <yann.eugone@gmail.com>
  */
-class OctoEnumExtension extends Extension
+class EnumExtension extends Extension
 {
     /**
      * {@inheritdoc}
diff --git a/EnumBundle.php b/EnumBundle.php
index 8892b78..bad4d50 100644
--- a/EnumBundle.php
+++ b/EnumBundle.php
@@ -3,6 +3,7 @@
 namespace EnumBundle;
 
 use EnumBundle\DependencyInjection\CompilerPass\CollectEnumCompilerPass;
+use EnumBundle\DependencyInjection\EnumExtension;
 use Symfony\Component\DependencyInjection\ContainerBuilder;
 use Symfony\Component\HttpKernel\Bundle\Bundle;
 
@@ -18,4 +19,12 @@ public function build(ContainerBuilder $container)
     {
         $container->addCompilerPass(new CollectEnumCompilerPass);
     }
+
+    /**
+     * {@inheritdoc}
+     */
+    public function getContainerExtension()
+    {
+        return new EnumExtension;
+    }
 }