From d015da113c0aaf877e3e2f126f39304ccc6c6086 Mon Sep 17 00:00:00 2001
From: Elijah <elijahahianyo@gmail.com>
Date: Fri, 25 Oct 2024 17:15:58 +0000
Subject: [PATCH] possible tests fix

---
 tests/units/components/test_props.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/tests/units/components/test_props.py b/tests/units/components/test_props.py
index 35880e352..8ab07f135 100644
--- a/tests/units/components/test_props.py
+++ b/tests/units/components/test_props.py
@@ -1,9 +1,13 @@
 import pytest
-from pydantic import ValidationError
 
 from reflex.components.props import NoExtrasAllowedProps
 from reflex.utils.exceptions import InvalidPropValueError
 
+try:
+    from pydantic.v1 import ValidationError
+except ModuleNotFoundError:
+    from pydantic import ValidationError
+
 
 class PropA(NoExtrasAllowedProps):
     """Base prop class."""