diff --git a/examples/oauth_api_server/server.py b/examples/oauth_api_server/server.py
index b58218b..d5190e5 100644
--- a/examples/oauth_api_server/server.py
+++ b/examples/oauth_api_server/server.py
@@ -37,7 +37,7 @@ def inner(*args, **kwargs):
             client = ps.connection['client']
 
             if session['token'] and hasattr(client, 'token') and client.token is None:
-                ps.set_token(session['token'])
+                client.set_token(session['token'])
 
         try:
             return f(*args, **kwargs)
diff --git a/flask_snow/__init__.py b/flask_snow/__init__.py
index 3ccc181..af953c2 100644
--- a/flask_snow/__init__.py
+++ b/flask_snow/__init__.py
@@ -12,7 +12,7 @@
 """
 
 __author__ = "Robert Wikman <rbw@vault13.org>"
-__version__ = "0.2.5"
+__version__ = "0.2.6"
 
 import warnings
 
@@ -183,4 +183,4 @@ def connection(self):
                     'resources': {}
                 }
 
-        return ctx.snow
+            return ctx.snow