lib/rubocop/cop/rails/not_null_column.rb in rubocop-rails-2.8.1 vs lib/rubocop/cop/rails/not_null_column.rb in rubocop-rails-2.9.0

@@ -14,11 +14,12 @@
       #   # good
       #   add_column :users, :name, :string, null: true
       #   add_column :users, :name, :string, null: false, default: ''
       #   add_reference :products, :category
       #   add_reference :products, :category, null: false, default: 1
-      class NotNullColumn < Cop
+      class NotNullColumn < Base
         MSG = 'Do not add a NOT NULL column without a default value.'
+        RESTRICT_ON_SEND = %i[add_column add_reference].freeze
 
         def_node_matcher :add_not_null_column?, <<~PATTERN
           (send nil? :add_column _ _ _ (hash $...))
         PATTERN