diff --git a/app/controllers/account_controller.rb b/app/controllers/account_controller.rb
index cb556f8d..eb4f1a82 100644
--- a/app/controllers/account_controller.rb
+++ b/app/controllers/account_controller.rb
@@ -377,10 +377,10 @@ class AccountController < ApplicationController
       else
         us = UsersService.new
         @user = us.register user_params.merge(:should_confirmation_password => false)
-        # # 注册时，记录是否是引流用户
-        # ip = request.remote_ip
-        # ua = UserAgent.find_by_ip(ip)
-        # ua.update_column(:type, UserAgent::REGISTER) if ua
+        # 注册时，记录是否是引流用户
+        ip = request.remote_ip
+        ua = UserAgent.find_by_ip(ip)
+        ua.update_column(:type, UserAgent::REGISTER) if ua
         if !@user.new_record?
           self.logged_user = @user
           flash[:notice] = l(:notice_account_activated)
diff --git a/app/controllers/competition_teams_controller.rb b/app/controllers/competition_teams_controller.rb
index b23ec5d6..9159cf07 100644
--- a/app/controllers/competition_teams_controller.rb
+++ b/app/controllers/competition_teams_controller.rb
@@ -114,9 +114,7 @@ class CompetitionTeamsController < ApplicationController
         new_team.team_members.create!(user_id: user_id, role: 3, competition_id: @competition.id, is_teacher: 1)
       end
       # 创建时，记录是否是引流用户
-      # ip = request.remote_ip
-      # ua = UserAgent.find_by_ip(ip)
-      # ua.update_column(:type, UserAgent::COMPETITION) if ua
+      record_agent_user_action
     end
   end
 
@@ -230,10 +228,8 @@ class CompetitionTeamsController < ApplicationController
       end
 
       team.team_members.create!(user_id: User.current.id, role: 2, competition_id: @competition.id)
-      # # 创建时，记录是否是引流用户
-      # ip = request.remote_ip
-      # ua = UserAgent.find_by_ip(ip)
-      # ua.update_column(:type, UserAgent::COMPETITION) if ua
+      # 记录引流
+      record_agent_user_action
     end
   end
 
@@ -351,7 +347,7 @@ class CompetitionTeamsController < ApplicationController
   end
 
   def record_agent_user_action
-    # 注册时，记录是否是引流用户
+    # 记录是否是引流用户的行为
     ip = request.remote_ip
     ua = UserAgent.find_by_ip(ip)
     ua.update_column(:type, UserAgent::COMPETITION) if ua