diff --git a/db/docs/subscription_user_add_on_assignments.yml b/db/docs/subscription_user_add_on_assignments.yml
index 38006808d3e7a09a97027238c393daa9ea0d6a81..65d59135f88d265b758afe7cda53e2e44a61ce07 100644
--- a/db/docs/subscription_user_add_on_assignments.yml
+++ b/db/docs/subscription_user_add_on_assignments.yml
@@ -8,4 +8,12 @@ feature_categories:
 classes:
 - GitlabSubscriptions::UserAddOnAssignment
 gitlab_schema: gitlab_main_cell
-sharding_key_issue_url: https://gitlab.com/gitlab-org/gitlab/-/issues/461396
+desired_sharding_key:
+  organization_id:
+    references: organizations
+    backfill_via:
+      parent:
+        foreign_key: add_on_purchase_id
+        table: subscription_add_on_purchases
+        sharding_key: organization_id
+        belongs_to: add_on_purchase