diff --git a/bundle/Controller/DownloadController.php b/bundle/Controller/DownloadController.php
index a30a714..104acb5 100644
--- a/bundle/Controller/DownloadController.php
+++ b/bundle/Controller/DownloadController.php
@@ -9,14 +9,12 @@
 use Netgen\Bundle\InformationCollectionBundle\Entity\EzInfoCollectionAttribute;
 use Netgen\Bundle\InformationCollectionBundle\Repository\EzInfoCollectionAttributeRepository;
 use Netgen\Bundle\InformationCollectionBundle\Repository\EzInfoCollectionRepository;
-use Symfony\Bundle\FrameworkBundle\Controller\ControllerTrait;
+use Symfony\Bundle\FrameworkBundle\Controller\AbstractController;
 use Symfony\Component\HttpFoundation\ResponseHeaderBag;
 use Symfony\Component\Security\Core\Exception\AccessDeniedException;
 
-class DownloadController
+class DownloadController extends AbstractController
 {
-    use ControllerTrait;
-
     private $infocollectionAttributeRepository;
 
     private $infocollectionRepository;
diff --git a/bundle/DependencyInjection/Configuration.php b/bundle/DependencyInjection/Configuration.php
index 4a7ec0c..5aae2e2 100644
--- a/bundle/DependencyInjection/Configuration.php
+++ b/bundle/DependencyInjection/Configuration.php
@@ -12,8 +12,8 @@ class Configuration implements ConfigurationInterface
      */
     public function getConfigTreeBuilder()
     {
-        $treeBuilder = new TreeBuilder();
-        $rootNode = $treeBuilder->root('netgen_enhanced_ez_binary_file');
+        $treeBuilder = new TreeBuilder('netgen_enhanced_ez_binary_file');
+        $rootNode = $treeBuilder->getRootNode();
 
         return $treeBuilder;
     }