diff --git a/lib/graphql_devise/mutations/login.rb b/lib/graphql_devise/mutations/login.rb
index e07e8aa5..c246615b 100644
--- a/lib/graphql_devise/mutations/login.rb
+++ b/lib/graphql_devise/mutations/login.rb
@@ -24,6 +24,8 @@ def resolve(email:, password:)
 
           yield resource if block_given?
 
+          context[:current_resource] = resource if context[:current_resource].nil?
+
           { authenticatable: resource, credentials: new_headers }
         elsif resource && !active_for_authentication?(resource)
           if locked?(resource)