lib/rom/repository/changeset/update.rb in rom-repository-1.0.0 vs lib/rom/repository/changeset/update.rb in rom-repository-1.0.1

@@ -1,5 +1,7 @@
+require 'rom/repository/changeset/restricted'
+
 module ROM
   class Changeset
     # Changeset specialization for update commands
     #
     # Update changesets will only execute their commands when
@@ -11,9 +13,11 @@
     #
     # @see Changeset::Stateful
     #
     # @api public
     class Update < Stateful
+      include Restricted
+
       command_type :update
 
       # Commit update changeset if there's a diff
       #
       # This returns original tuple if there's no diff