diff --git a/argos/schemas/config.py b/argos/schemas/config.py
index 4d78cf1..486af14 100644
--- a/argos/schemas/config.py
+++ b/argos/schemas/config.py
@@ -1,6 +1,6 @@
 from typing import Dict, List, Literal, Optional, Tuple
 
-from pydantic import BaseModel, HttpUrl, field_validator
+from pydantic import BaseModel, ConfigDict, HttpUrl, field_validator
 from pydantic.functional_validators import BeforeValidator
 from typing_extensions import Annotated
 
@@ -26,9 +26,7 @@ class SSL(BaseModel):
 class WebsiteCheck(BaseModel):
     key: str
     value: str | List[str] | Dict[str, str]
-
-    class Config:
-        arbitrary_types_allowed = True
+    model_config = ConfigDict(arbitrary_types_allowed=True)
 
     @classmethod
     def __get_validators__(cls):
diff --git a/argos/schemas/models.py b/argos/schemas/models.py
index f1e1bbb..c3d0e82 100644
--- a/argos/schemas/models.py
+++ b/argos/schemas/models.py
@@ -2,7 +2,7 @@ import traceback
 from datetime import datetime
 from typing import Literal
 
-from pydantic import BaseModel
+from pydantic import BaseModel, ConfigDict
 
 # XXX Refactor using SQLModel to avoid duplication of model data
 
@@ -16,8 +16,7 @@ class Task(BaseModel):
     selected_at: datetime | None
     selected_by: str | None
 
-    class Config:
-        from_attributes = True
+    model_config = ConfigDict(from_attributes=True)
 
     def __str__(self):
         id = self.id