diff --git a/geomagio/algorithm/MetadataAlgorithm.py b/geomagio/algorithm/MetadataAlgorithm.py
index 027e753d02fbc83b578443872a8c44fbfb474a0f..43b96804c9bd8f94fc039e00016ff9dfd6a196b0 100644
--- a/geomagio/algorithm/MetadataAlgorithm.py
+++ b/geomagio/algorithm/MetadataAlgorithm.py
@@ -4,7 +4,6 @@ from datetime import timedelta
 from enum import Enum
 from typing import Optional
 
-from ..metadata.flag.Flag import Flag
 from ..metadata import Metadata, MetadataFactory, MetadataCategory
 from ..edge.MiniSeedFactory import MiniSeedFactory
 from ..edge.EdgeFactory import EdgeFactory
@@ -55,7 +54,7 @@ class MetadataAlgorithm(BaseModel):
 
     def create_metadata(
         self,
-        metadata_class: Flag,
+        metadata_class: MetadataCategory,
         metadata_dict: dict,
         category: MetadataCategory,
         network: str,
@@ -103,7 +102,7 @@ class MetadataAlgorithm(BaseModel):
         return self._get_metadata_factory().create_metadata(metadata=metadata_obj)
 
     def split_stream_by_day(self, stream: Stream) -> list[Stream]:
-        """Split stream into daily streams to prevent metadata from overlapping."""
+        """Split stream into daily streams to control size of spike arrays."""
         daily_streams = []
         # get min and max time
         current_time = min(trace.stats.starttime for trace in stream)