diff --git a/src/index.tsx b/src/index.tsx
index 4cab2ef..378b0b5 100644
--- a/src/index.tsx
+++ b/src/index.tsx
@@ -906,7 +906,7 @@ export type MutationSetLoggerLevelArgs = {
 
 export type MutationSetSegmentDurationArgs = {
   duration: Scalars["Float"]["input"];
-  segmentId: Scalars["Int"]["input"];
+  segmentIndex: Scalars["Int"]["input"];
   videoId: Scalars["Int"]["input"];
 };
 
diff --git a/src/schema.gql b/src/schema.gql
index 2512dcc..a1ea6fd 100644
--- a/src/schema.gql
+++ b/src/schema.gql
@@ -365,7 +365,11 @@ type Mutation {
   ): CreateUploadStreamReturn!
   getUploadLink(videoId: Int!, segmentIndex: Int!): GetUploadLinkReturn!
   getHlsInitUploadLink(videoId: Int!): GetUploadLinkReturn!
-  setSegmentDuration(videoId: Int!, segmentId: Int!, duration: Float!): Boolean!
+  setSegmentDuration(
+    videoId: Int!
+    segmentIndex: Int!
+    duration: Float!
+  ): Boolean!
   editUploadStream(videoId: Int!, videoMetadata: VideoMetadataInput!): Boolean!
   deleteVideo(videoId: Int!): Boolean!
 }