diff --git a/zeno/agents/layerfinder/agent.py b/zeno/agents/layerfinder/agent.py
index 0be7a1a..96959d0 100644
--- a/zeno/agents/layerfinder/agent.py
+++ b/zeno/agents/layerfinder/agent.py
@@ -6,13 +6,7 @@
 from typing_extensions import Self
 
 
-# Add this mixin to all models
-class JSONSerializable(BaseModel):
-    def to_dict(self):
-        return json.loads(json.dumps(self, default=lambda o: o.dict()))
-
-
-class Dataset(JSONSerializable):
+class Dataset(BaseModel):
     title: str = Field(description="The title of the dataset")
     dataset: str = Field(description="The slug of the dataset")
     explanation: str = Field(
@@ -31,7 +25,7 @@ def set_uri_and_tilelayer(self) -> Self:
         return self
 
 
-class LayerFinderResponse(JSONSerializable):
+class LayerFinderResponse(BaseModel):
     datasets: List[Dataset]