diff --git a/src/dreamii/python.py b/src/dreamii/python.py
index 3c02080..c9e5ee2 100644
--- a/src/dreamii/python.py
+++ b/src/dreamii/python.py
@@ -123,7 +123,7 @@ class PythonExtension(Extension):
                     print(html.escape(result.stderr) if escape else result.stderr, end="")
 
             return html.escape(result.stdout + result.stderr) if stderr else result.stdout
-
+        environment = self.environment
         db = self.app.db
         app = self.app
         dreamii = self.app
@@ -133,4 +133,4 @@ class PythonExtension(Extension):
         result = sys.stdout.getvalue()
         if not result:
             result = sys.stderr.getvalue()
-        return result
\ No newline at end of file
+        return result