diff --git a/src/ai_models/outputs/__init__.py b/src/ai_models/outputs/__init__.py
index 20874b2..fcf3ea4 100644
--- a/src/ai_models/outputs/__init__.py
+++ b/src/ai_models/outputs/__init__.py
@@ -25,7 +25,7 @@ def finalise(self, *args, **kwargs):
         pass
 
 
-class FileOutputBase(Output):
+class GribOutputBase(Output):
     def __init__(self, owner, path, metadata, **kwargs):
         self._first = True
         metadata.setdefault("stream", "oper")
@@ -93,7 +93,7 @@ def write(self, data, *args, check=False, **kwargs):
         return handle, path
 
 
-class FileOutput(FileOutputBase):
+class FileOutput(GribOutputBase):
     def __init__(self, *args, **kwargs):
         super().__init__(*args, **kwargs)
         LOG.info("Writing results to %s", self.path)