From fb65d09e9fd359e7ef636ce3a755e21eac6240c7 Mon Sep 17 00:00:00 2001
From: phuongntt <phuongntt@cystack.net>
Date: Tue, 13 Feb 2024 09:53:48 +0700
Subject: [PATCH] chore: Fix bug referrals

---
 locker_server/api/v1_0/referrals/views.py | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/locker_server/api/v1_0/referrals/views.py b/locker_server/api/v1_0/referrals/views.py
index adadd324..0eb81e68 100644
--- a/locker_server/api/v1_0/referrals/views.py
+++ b/locker_server/api/v1_0/referrals/views.py
@@ -38,11 +38,14 @@ def claim(self, request, *args, **kwargs):
 
         # Upgrade current user to Premium
         current_time = now()
-        self.user_service.update_plan(user=user, plan_type_alias=PLAN_TYPE_PM_PREMIUM, **{
-            "start_period": current_time,
-            "end_period": current_time + 30 * 86400,
-            "cancel_at_period_end": True
-        })
+        self.user_service.update_plan(
+            user_id=user.user_id, plan_type_alias=PLAN_TYPE_PM_PREMIUM,
+            **{
+                "start_period": current_time,
+                "end_period": current_time + 30 * 86400,
+                "cancel_at_period_end": True
+            }
+        )
         # Get current plan of the referred user
         if count < REFERRAL_LIMIT:
             referred_user_plan = self.user_service.get_current_plan(