diff --git a/app/dashboard/views/pricing.py b/app/dashboard/views/pricing.py
index 27abb572..7c8f1ac7 100644
--- a/app/dashboard/views/pricing.py
+++ b/app/dashboard/views/pricing.py
@@ -62,7 +62,7 @@ def pricing():
                 "error",
             )
             return redirect(url_for("dashboard.index"))
-        proton_upgrade = partner_user.id == get_proton_partner().id
+        proton_upgrade = partner_user.partner_id == get_proton_partner().id
 
     return render_template(
         "dashboard/pricing.html",