diff --git a/src/index.tsx b/src/index.tsx
index 6e2297b..5a58097 100644
--- a/src/index.tsx
+++ b/src/index.tsx
@@ -2404,15 +2404,6 @@ export type SuccessfulAddAddShotAnnotationErrors =
   | AddShotAnnotationErrors
   | SuccessfulAdd;
 
-export type SuccessfulUpdate = {
-  __typename?: "SuccessfulUpdate";
-  value: Scalars["Boolean"]["output"];
-};
-
-export type SuccessfulUpdateUpdateShotAnnotationErrors =
-  | SuccessfulUpdate
-  | UpdateShotAnnotationErrors;
-
 export type TagGql = {
   __typename?: "TagGQL";
   group?: Maybe<Scalars["String"]["output"]>;
@@ -2455,14 +2446,10 @@ export type UpdateAnnotationInputGql = {
   notes?: InputMaybe<Scalars["String"]["input"]>;
 };
 
-export type UpdateShotAnnotationErrors = {
-  __typename?: "UpdateShotAnnotationErrors";
-  error?: Maybe<DoesNotOwnShotErr>;
-};
-
 export type UpdateShotAnnotationReturn = {
   __typename?: "UpdateShotAnnotationReturn";
-  value: SuccessfulUpdateUpdateShotAnnotationErrors;
+  error?: Maybe<DoesNotOwnShotErr>;
+  shot?: Maybe<ShotGql>;
 };
 
 export type UploadLink = {
@@ -2768,16 +2755,26 @@ export type UpdateShotAnnotationsMutation = {
   __typename?: "Mutation";
   updateShotAnnotations: {
     __typename?: "UpdateShotAnnotationReturn";
-    value:
-      | { __typename?: "SuccessfulUpdate"; value: boolean }
-      | {
-          __typename?: "UpdateShotAnnotationErrors";
-          error?: {
-            __typename?: "DoesNotOwnShotErr";
-            shotId: number;
-            msg?: string | null;
-          } | null;
+    shot?: {
+      __typename?: "ShotGQL";
+      id: number;
+      annotations: Array<{
+        __typename?: "ShotAnnotationGQL";
+        shotId: number;
+        notes: string;
+        errorDefault: boolean;
+        type: {
+          __typename?: "ShotAnnotationTypeGQL";
+          id: number;
+          name: string;
         };
+      }>;
+    } | null;
+    error?: {
+      __typename?: "DoesNotOwnShotErr";
+      shotId: number;
+      msg?: string | null;
+    } | null;
   };
 };
 
@@ -4340,17 +4337,22 @@ export const UpdateShotAnnotationsDocument = gql`
     $annotations: [UpdateAnnotationInputGQL!]!
   ) {
     updateShotAnnotations(shotId: $shotId, annotations: $annotations) {
-      value {
-        ... on SuccessfulUpdate {
-          value
-        }
-        ... on UpdateShotAnnotationErrors {
-          error {
-            shotId
-            msg
+      shot {
+        id
+        annotations {
+          shotId
+          type {
+            id
+            name
           }
+          notes
+          errorDefault
         }
       }
+      error {
+        shotId
+        msg
+      }
     }
   }
 `;
diff --git a/src/operations/shots.gql b/src/operations/shots.gql
index 1d97898..9aa6e65 100644
--- a/src/operations/shots.gql
+++ b/src/operations/shots.gql
@@ -22,17 +22,22 @@ mutation UpdateShotAnnotations(
   $annotations: [UpdateAnnotationInputGQL!]!
 ) {
   updateShotAnnotations(shotId: $shotId, annotations: $annotations) {
-    value {
-      ... on SuccessfulUpdate {
-        value
-      }
-      ... on UpdateShotAnnotationErrors {
-        error {
-          shotId
-          msg
+    shot {
+      id
+      annotations {
+        shotId
+        type {
+          id
+          name
         }
+        notes
+        errorDefault
       }
     }
+    error {
+      shotId
+      msg
+    }
   }
 }
 
diff --git a/src/schema.gql b/src/schema.gql
index 03162f7..e0d78c8 100644
--- a/src/schema.gql
+++ b/src/schema.gql
@@ -650,18 +650,7 @@ type OtherErrorNeedsNote {
 }
 
 type UpdateShotAnnotationReturn {
-  value: SuccessfulUpdateUpdateShotAnnotationErrors!
-}
-
-union SuccessfulUpdateUpdateShotAnnotationErrors =
-    SuccessfulUpdate
-  | UpdateShotAnnotationErrors
-
-type SuccessfulUpdate {
-  value: Boolean!
-}
-
-type UpdateShotAnnotationErrors {
+  shot: ShotGQL
   error: DoesNotOwnShotErr
 }