diff --git a/src/fiber_package/membership_tag.py b/src/fiber_package/membership_tag.py
index 5f32636..119e947 100644
--- a/src/fiber_package/membership_tag.py
+++ b/src/fiber_package/membership_tag.py
@@ -18,7 +18,7 @@ def validate_visibility_tag(v: str) -> str:
         return v
     raise ValueError(f"Invalid visibility tag: {v!r}")
 
-VisibilityTag = Annotated[str, AfterValidator(validate_visibility_tag)]
+MembershipTag = Annotated[str, AfterValidator(validate_visibility_tag)]
 
-def generate_tag(association_tag: str, membership_tag: MembershipTag):
+def generate_tag(association_tag: str, membership: Membership) -> MembershipTag:
     return association_tag + ":" + membership_tag