From e5315dc016ddee35676513b9bbf8e1899e5fff4c Mon Sep 17 00:00:00 2001
From: Alex Bilbie <hello@alexbilbie.com>
Date: Thu, 1 May 2014 14:57:12 +0100
Subject: [PATCH] Test fixes

---
 src/Entity/Session.php          | 2 +-
 src/Storage/ClientInterface.php | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/Entity/Session.php b/src/Entity/Session.php
index 767403c4..ab166253 100644
--- a/src/Entity/Session.php
+++ b/src/Entity/Session.php
@@ -209,7 +209,7 @@ class Session
             return $this->client;
         }
 
-        $this->client = $this->server->getStorage('client')->getBySession($this->getId());
+        $this->client = $this->server->getStorage('client')->getBySession($this);
         return $this->client;
     }
 
diff --git a/src/Storage/ClientInterface.php b/src/Storage/ClientInterface.php
index fa881356..267888f0 100644
--- a/src/Storage/ClientInterface.php
+++ b/src/Storage/ClientInterface.php
@@ -11,6 +11,8 @@
 
 namespace League\OAuth2\Server\Storage;
 
+use League\OAuth2\Server\Entity\Session;
+
 /**
  * Client storage interface
  */