diff --git a/clients/client-python/gravitino/client/fileset_catalog.py b/clients/client-python/gravitino/client/fileset_catalog.py
index df0ecf579e2..39a095ba9ac 100644
--- a/clients/client-python/gravitino/client/fileset_catalog.py
+++ b/clients/client-python/gravitino/client/fileset_catalog.py
@@ -178,7 +178,7 @@ def create_fileset(
         fileset_resp = FilesetResponse.from_json(resp.body, infer_missing=True)
         fileset_resp.validate()
 
-        return fileset_resp.fileset()
+        return GenericFileset(fileset_resp.fileset(), self.rest_client, full_namespace)
 
     def alter_fileset(self, ident: NameIdentifier, *changes) -> Fileset:
         """Update a fileset metadata in the catalog.
@@ -187,7 +187,7 @@ def alter_fileset(self, ident: NameIdentifier, *changes) -> Fileset:
             ident: A fileset identifier, which should be "schema.fileset" format.
             changes: The changes to apply to the fileset.
 
-        Args:
+        Raises:
             IllegalArgumentException If the changes are invalid.
             NoSuchFilesetException If the fileset does not exist.
 
@@ -212,7 +212,7 @@ def alter_fileset(self, ident: NameIdentifier, *changes) -> Fileset:
         fileset_resp = FilesetResponse.from_json(resp.body, infer_missing=True)
         fileset_resp.validate()
 
-        return fileset_resp.fileset()
+        return GenericFileset(fileset_resp.fileset(), self.rest_client, full_namespace)
 
     def drop_fileset(self, ident: NameIdentifier) -> bool:
         """Drop a fileset from the catalog.