diff --git a/lib/graphiti/resource/persistence.rb b/lib/graphiti/resource/persistence.rb
index 1f7b4b0c..5a1c61a9 100644
--- a/lib/graphiti/resource/persistence.rb
+++ b/lib/graphiti/resource/persistence.rb
@@ -4,11 +4,11 @@ module Persistence
       extend ActiveSupport::Concern
 
       class_methods do
-        def before_attributes(method = nil, only: [:create, :update, :assign], &blk)
+        def before_attributes(method = nil, only: [:create, :update], &blk)
           add_callback(:attributes, :before, method, only, &blk)
         end
 
-        def after_attributes(method = nil, only: [:create, :update, :assign], &blk)
+        def after_attributes(method = nil, only: [:create, :update], &blk)
           add_callback(:attributes, :after, method, only, &blk)
         end