diff --git a/Resources/config/templates/knockout.yml b/Resources/config/templates/knockout.yml
index c382a21..2d02ccf 100644
--- a/Resources/config/templates/knockout.yml
+++ b/Resources/config/templates/knockout.yml
@@ -136,7 +136,7 @@ parameters:
             filename: 'DependencyInjection/Configuration.php'
             template: 'AvroGeneratorBundle:Templates/Knockout/DependencyInjection/Configuration.php'
         extension:
-            filename: 'DependencyInjection/{{ bundleVendor }}/{{ bundleCoreName }}Extension.php'
+            filename: 'DependencyInjection/{{ bundleVendor }}{{ bundleCoreName }}Extension.php'
             template: 'AvroGeneratorBundle:Templates/Knockout/DependencyInjection/Extension.php'
         readme:
             filename: 'README.md'
diff --git a/Templates/Knockout/DependencyInjection/Extension.php b/Templates/Knockout/DependencyInjection/Extension.php
index ecfe0c3..e108b10 100644
--- a/Templates/Knockout/DependencyInjection/Extension.php
+++ b/Templates/Knockout/DependencyInjection/Extension.php
@@ -4,6 +4,7 @@
 use Symfony\Component\HttpKernel\DependencyInjection\Extension;
 use Symfony\Component\DependencyInjection\ContainerBuilder;
 use Symfony\Component\DependencyInjection\Loader\YamlFileLoader;
+use Symfony\Component\Config\Definition\Processor;
 use Symfony\Component\Config\FileLocator;
 
 class {{ bundleVendor }}{{ bundleCoreName | ucFirst }}Extension extends Extension