diff --git a/Entity/XhguiRuns.php b/Entity/XhguiRuns.php
index a99dafd..2f71e9d 100644
--- a/Entity/XhguiRuns.php
+++ b/Entity/XhguiRuns.php
@@ -4,11 +4,24 @@
 
 use iXHProfRuns;
 use Symfony\Component\DependencyInjection\ContainerAwareInterface;
-use Symfony\Component\DependencyInjection\ContainerAwareTrait;
+use Symfony\Component\DependencyInjection\ContainerInterface;
 
 class XhguiRuns implements iXHProfRuns, ContainerAwareInterface
 {
-    use ContainerAwareTrait;
+    /**
+     * @var ContainerInterface
+     */
+    protected $container;
+
+    /**
+     * Sets the Container associated with this Controller.
+     *
+     * @param ContainerInterface $container A ContainerInterface instance
+     */
+    public function setContainer(ContainerInterface $container = null)
+    {
+        $this->container = $container;
+    }
 
     public function __construct($serverName, $uri) {
     	$this->serverName = $serverName;