From e23581e260ae21e67eae438cd38f0fb77ea9a6e7 Mon Sep 17 00:00:00 2001
From: Matt Wright <mdw1980@gmail.com>
Date: Mon, 16 Jul 2012 19:34:33 -0400
Subject: [PATCH] Polish

---
 flask_security/core.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/flask_security/core.py b/flask_security/core.py
index d0a632d..9b776ae 100644
--- a/flask_security/core.py
+++ b/flask_security/core.py
@@ -232,14 +232,11 @@ class Security(object):
                                         url_prefix=cv('URL_PREFIX', app=app))
             app.register_blueprint(bp)
 
-        if not hasattr(app, 'extensions'):
-            app.extensions = {}
-
         kwargs = {}
         for key, value in get_config(app).items():
             kwargs[key.lower()] = value
 
-        app.extensions['security'] = _SecurityState(
+        state = _SecurityState(
             app=app,
             datastore=datastore,
             auth_provider=AuthenticationProvider(),
@@ -251,6 +248,11 @@ class Security(object):
             token_auth_serializer=_get_token_auth_serializer(app),
             **kwargs)
 
+        if not hasattr(app, 'extensions'):
+            app.extensions = {}
+
+        app.extensions['security'] = state
+
 
 class AuthenticationProvider(object):
     """The default authentication provider implementation."""