diff --git a/app/api/mobile/apis/memos.rb b/app/api/mobile/apis/memos.rb
index 89412316..a72b9b6c 100644
--- a/app/api/mobile/apis/memos.rb
+++ b/app/api/mobile/apis/memos.rb
@@ -33,7 +33,7 @@ module Mobile
           if params[:edu]
             Rails.logger.info("####11111222##{@env['HTTP_X_REAL_IP']}")
             ip = @env['HTTP_X_REAL_IP']
-            ua = UserAgent.where(:ip => ip)
+            ua = UserAgent.find_by_ip(ip)
             if ua.blank?
               UserAgent.create!(:key => params[:edu].strip, :ip => ip, :type => UserAgent::AD)
             end
diff --git a/app/controllers/account_controller.rb b/app/controllers/account_controller.rb
index d362cde3..eb4f1a82 100644
--- a/app/controllers/account_controller.rb
+++ b/app/controllers/account_controller.rb
@@ -379,7 +379,7 @@ class AccountController < ApplicationController
         @user = us.register user_params.merge(:should_confirmation_password => false)
         # 注册时，记录是否是引流用户
         ip = request.remote_ip
-        ua = UserAgent.find_by_id(:ip => ip)
+        ua = UserAgent.find_by_ip(ip)
         ua.update_column(:type, UserAgent::REGISTER) if ua
         if !@user.new_record?
           self.logged_user = @user
diff --git a/app/controllers/competition_teams_controller.rb b/app/controllers/competition_teams_controller.rb
index 092426ef..3efaa1cf 100644
--- a/app/controllers/competition_teams_controller.rb
+++ b/app/controllers/competition_teams_controller.rb
@@ -349,7 +349,7 @@ class CompetitionTeamsController < ApplicationController
   def record_agent_user_action
     # 注册时，记录是否是引流用户
     ip = request.remote_ip
-    ua = UserAgent.find_by_id(:ip => ip)
+    ua = UserAgent.find_by_ip(ip)
     ua.update_column(:type, UserAgent::COMPETITION) if ua
   end
 end