2 files changed, 2 insertions, 2 deletions
diff --git a/lib/arel/crud.rb b/lib/arel/crud.rb
index e6b1cd9675..bedfb8c30c 100644
--- a/lib/arel/crud.rb
+++ b/lib/arel/crud.rb
@@ -12,7 +12,7 @@ module Arel
       end
       um.table relation
       um.set values
-      um.take @ast.limit
+      um.take @ast.limit.expr if @ast.limit
       um.order(*@ast.orders)
       um.wheres = @ctx.wheres
       um
diff --git a/lib/arel/update_manager.rb b/lib/arel/update_manager.rb
index cf24dbac92..f13aeb0a8c 100644
--- a/lib/arel/update_manager.rb
+++ b/lib/arel/update_manager.rb
@@ -7,7 +7,7 @@ module Arel
     end
 
     def take limit
-      @ast.limit = limit
+      @ast.limit = Nodes::Limit.new(limit) if limit
       self
     end