1 files changed, 3 insertions, 2 deletions
diff --git a/lib/arel/select_manager.rb b/lib/arel/select_manager.rb
index 274633c9db..3e0d4038d7 100644
--- a/lib/arel/select_manager.rb
+++ b/lib/arel/select_manager.rb
@@ -10,7 +10,7 @@ module Arel
     end
 
     def limit
-      @ast.limit
+      @ast.limit && @ast.limit.expr
     end
     alias :taken :limit
 
@@ -137,7 +137,8 @@ module Arel
     end
 
     def take limit
-      @ast.limit = limit
+      @ast.limit = Nodes::Limit.new(limit)
+      @ctx.top   = Nodes::Top.new(limit)
       self
     end