diff --git a/Core/PageTree/AlPageTree.php b/Core/PageTree/AlPageTree.php
index <HASH>..<HASH> 100644
--- a/Core/PageTree/AlPageTree.php
+++ b/Core/PageTree/AlPageTree.php
@@ -59,7 +59,7 @@ class AlPageTree
     /**
      * Constructor
      *
-     * @param ContainerInterface $container
+     * @param Symfony\Component\DependencyInjection\ContainerInterface $container
      * @param AlPageBlocksInterface $pageBlocks
      */
     public function __construct(ContainerInterface $container, AlPageBlocksInterface $pageBlocks = null)
@@ -68,6 +68,16 @@ class AlPageTree
         $this->pageBlocks = $pageBlocks;
         $this->activeTheme = $this->container->get('alphalemon_theme_engine.active_theme');
     }
+    
+    /**
+     * Returns the container 
+     * 
+     * @return Symfony\Component\DependencyInjection\ContainerInterface
+     */
+    public function getContainer()
+    {
+        return $this->container;
+    }
 
     /**
      * Sets the template object
Added method to retrieve the container object