diff --git a/packages/ui/src/app/App.stories.tsx b/packages/ui/src/app/App.stories.tsx
index 23008abecd..2f7343de73 100644
--- a/packages/ui/src/app/App.stories.tsx
+++ b/packages/ui/src/app/App.stories.tsx
@@ -25,6 +25,7 @@ type Args = {
   hasWallet: boolean
   isRPCNodeConnected: boolean
   onBuyMembership: CallableFunction
+  onUpdateMembership: CallableFunction
   onTransfer: CallableFunction
 }
 
@@ -107,6 +108,12 @@ export default {
                     onSend: args.onBuyMembership,
                     failure: parameters.txFailure,
                   },
+                  updateProfile: {
+                    event: 'MembershipUpdated',
+                    data: [MEMBER_DATA.id],
+                    onSend: args.onUpdateMembership,
+                    failure: parameters.txFailure,
+                  },
                 },
               },
             },
diff --git a/packages/ui/src/memberships/modals/BuyMembershipModal/machine.ts b/packages/ui/src/memberships/modals/BuyMembershipModal/machine.ts
index b3054ac664..4b56fcdff7 100644
--- a/packages/ui/src/memberships/modals/BuyMembershipModal/machine.ts
+++ b/packages/ui/src/memberships/modals/BuyMembershipModal/machine.ts
@@ -108,9 +108,6 @@ export const buyMembershipMachine = createMachine<BuyMembershipContext, BuyMembe
         onDone: [
           {
             target: 'success',
-            // actions: assign({
-            //   memberId: (context, event) => getDataFromEvent(event.data.events, 'members', 'MembershipBought', 0),
-            // }),
             cond: isTransactionSuccess,
           },
           {