2 files changed, 4 insertions, 4 deletions
diff --git a/lib/arel/visitors/mysql.rb b/lib/arel/visitors/mysql.rb
index 8f9c3cc809..dad4d5e3b2 100644
--- a/lib/arel/visitors/mysql.rb
+++ b/lib/arel/visitors/mysql.rb
@@ -3,8 +3,8 @@ module Arel
     class MySQL < Arel::Visitors::ToSql
       private
       def visit_Arel_Nodes_Lock o
-        if o.locking.is_a?(String)
-          o.locking
+        if o.expr.is_a?(String)
+          o.expr
         else
           "FOR UPDATE"
         end
diff --git a/lib/arel/visitors/postgresql.rb b/lib/arel/visitors/postgresql.rb
index 45e7349fd3..68b483910e 100644
--- a/lib/arel/visitors/postgresql.rb
+++ b/lib/arel/visitors/postgresql.rb
@@ -3,8 +3,8 @@ module Arel
     class PostgreSQL < Arel::Visitors::ToSql
       private
       def visit_Arel_Nodes_Lock o
-        if o.locking.is_a?(String)
-          o.locking
+        if o.expr.is_a?(String)
+          o.expr
         else
           "FOR UPDATE"
         end