diff --git a/src/SASViyaApiClient.ts b/src/SASViyaApiClient.ts
index 6f597c5..bb8d00c 100644
--- a/src/SASViyaApiClient.ts
+++ b/src/SASViyaApiClient.ts
@@ -323,7 +323,11 @@ export class SASViyaApiClient {
     const updateContextRequest: RequestInit = {
       method: 'PUT',
       headers,
-      body: JSON.stringify({ ...context, ...updatedContext })
+      body: JSON.stringify({
+        ...context,
+        ...updatedContext,
+        attributes: { ...context.attributes, ...updatedContext.attributes }
+      })
     }
 
     return await this.request<Context>(
diff --git a/src/types/Context.ts b/src/types/Context.ts
index baa1da0..26e942c 100644
--- a/src/types/Context.ts
+++ b/src/types/Context.ts
@@ -3,6 +3,7 @@ export interface Context {
   id: string
   createdBy: string
   version: number
+  attributes?: any
 }
 
 export interface EditContextInput {
@@ -10,6 +11,7 @@ export interface EditContextInput {
   description?: string
   launchContext?: { name: string }
   environment?: { options?: string[]; autoExecLines?: string[] }
+  attributes?: any
   authorizedUsers?: string[]
   authorizeAllAuthenticatedUsers?: boolean
 }