diff --git a/src/workos/types/directory_sync/directory_user.py b/src/workos/types/directory_sync/directory_user.py
index 387948fa..ee710b0f 100644
--- a/src/workos/types/directory_sync/directory_user.py
+++ b/src/workos/types/directory_sync/directory_user.py
@@ -1,8 +1,7 @@
 from typing import Any, Dict, Literal, Optional, Sequence, Union
 
-from workos.types.workos_model import WorkOSModel
 from workos.types.directory_sync.directory_group import DirectoryGroup
-
+from workos.types.workos_model import WorkOSModel
 
 DirectoryUserState = Literal["active", "inactive"]
 
@@ -33,7 +32,7 @@ class DirectoryUser(WorkOSModel):
     # @deprecated Will be removed in a future major version. Enable the `username` custom attribute in dashboard and pull from customAttributes instead. See https://workos.com/docs/directory-sync/attributes/custom-attributes/auto-mapped-attributes for details.
     username: Optional[str] = None
     state: DirectoryUserState
-    custom_attributes: Dict[str, Any]
+    custom_attributes: Optional[Dict[str, Any]] = None
     raw_attributes: Dict[str, Any]
     created_at: str
     updated_at: str
diff --git a/src/workos/types/user_management/organization_membership.py b/src/workos/types/user_management/organization_membership.py
index 5c7bda0f..a89b15ab 100644
--- a/src/workos/types/user_management/organization_membership.py
+++ b/src/workos/types/user_management/organization_membership.py
@@ -1,4 +1,5 @@
 from typing import Any, Literal, Mapping, Optional, Sequence
+
 from typing_extensions import TypedDict
 
 from workos.types.workos_model import WorkOSModel
@@ -21,6 +22,6 @@ class OrganizationMembership(WorkOSModel):
     role: OrganizationMembershipRole
     roles: Optional[Sequence[OrganizationMembershipRole]] = None
     status: LiteralOrUntyped[OrganizationMembershipStatus]
-    custom_attributes: Mapping[str, Any]
+    custom_attributes: Optional[Mapping[str, Any]] = None
     created_at: str
     updated_at: str