diff --git a/stubs/ci3/libraries/Webguard.php b/stubs/ci3/libraries/Webguard.php
index 476a5e5..a9d9bee 100644
--- a/stubs/ci3/libraries/Webguard.php
+++ b/stubs/ci3/libraries/Webguard.php
@@ -37,8 +37,9 @@ public function guest()
     public function user()
     {
         $unserialize_session = unserialize($_SESSION['serialize_session']);
-        $this->user->roles = $unserialize_session['roles'];
-        $this->user->role = $unserialize_session['role'];
+        foreach ($unserialize_session as $key => $value) {
+            $this->user->{$key} = $value;
+        }
         return $this;
     }