diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 463a2a7..51ad9fc 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -21,12 +21,13 @@ repos:
         pass_filenames: false
         additional_dependencies: [
             pytest,
-            pydantic,
+            pydantic==2.7.4,
             types-Pillow,
             types-requests,
             types-pytz,
             types-setuptools,
             types-urllib3,
             types-aiofiles,
-            StrEnum
+            StrEnum,
+            horde_model_reference==0.8.1,
         ]
diff --git a/requirements.txt b/requirements.txt
index d6d29b6..1e9eff4 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
-horde_model_reference~=0.7.0
+horde_model_reference~=0.8.1
 
-pydantic
+pydantic==2.7.4
 requests
 StrEnum
 loguru
diff --git a/tests/ai_horde_api/test_ai_worker_roundtrip_api_calls.py b/tests/ai_horde_api/test_ai_worker_roundtrip_api_calls.py
index f43de89..2a6fb8d 100644
--- a/tests/ai_horde_api/test_ai_worker_roundtrip_api_calls.py
+++ b/tests/ai_horde_api/test_ai_worker_roundtrip_api_calls.py
@@ -1,47 +1,27 @@
 import asyncio
-import base64
-from typing import Any
 
 import aiohttp
 import PIL.Image
 import pytest
-from loguru import logger
 import yarl
+from loguru import logger
 
 from horde_sdk.ai_horde_api.ai_horde_clients import (
     AIHordeAPIAsyncClientSession,
     AIHordeAPIAsyncSimpleClient,
-    AIHordeAPISimpleClient,
 )
 from horde_sdk.ai_horde_api.apimodels import (
-    KNOWN_ALCHEMY_TYPES,
-    AlchemyAsyncRequest,
-    AlchemyAsyncRequestFormItem,
-    AlchemyStatusResponse,
     ImageGenerateAsyncRequest,
-    ImageGenerateAsyncResponse,
-    ImageGenerateCheckResponse,
-    ImageGenerateJobPopPayload,
     ImageGenerateJobPopRequest,
     ImageGenerateJobPopResponse,
-    ImageGenerateJobPopSkippedStatus,
     ImageGenerateStatusResponse,
-    ImageGenerationInputPayload,
     ImageGenerationJobSubmitRequest,
     JobSubmitResponse,
-    LorasPayloadEntry,
 )
-from horde_sdk.ai_horde_api.apimodels.base import ExtraSourceImageEntry
 from horde_sdk.ai_horde_api.consts import (
     GENERATION_STATE,
-    KNOWN_FACEFIXERS,
-    KNOWN_MISC_POST_PROCESSORS,
-    KNOWN_SOURCE_PROCESSING,
-    KNOWN_UPSCALERS,
-    POST_PROCESSOR_ORDER_TYPE,
 )
 from horde_sdk.ai_horde_api.fields import JobID
-from horde_sdk.generic_api.apimodels import RequestErrorResponse
 
 
 class TestImageWorkerRoundtrip:
diff --git a/tests/conftest.py b/tests/conftest.py
index dda7e2e..bdec2f1 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -2,8 +2,8 @@
 import os
 import pathlib
 
-from loguru import logger
 import pytest
+from loguru import logger
 
 os.environ["TESTS_ONGOING"] = "1"