From 77930bae16b89cb6aa692147a38c5a19bf77a7be Mon Sep 17 00:00:00 2001
From: Andrew Bell <115623869+andybharness@users.noreply.github.com>
Date: Mon, 22 Apr 2024 17:56:29 +0100
Subject: [PATCH] fixing imports

---
 tests/unit/test_clause.py    | 2 +-
 tests/unit/test_evaluator.py | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/unit/test_clause.py b/tests/unit/test_clause.py
index 5320e9d..3b11bcf 100644
--- a/tests/unit/test_clause.py
+++ b/tests/unit/test_clause.py
@@ -1,6 +1,6 @@
 import pytest
 
-from featureflags.evaluations.clause import Clause
+from featureflags.openapi.config.models.clause import Clause
 from featureflags.evaluations.constants import (CONTAINS_OPERATOR,
                                                 ENDS_WITH_OPERATOR,
                                                 EQUAL_OPERATOR,
diff --git a/tests/unit/test_evaluator.py b/tests/unit/test_evaluator.py
index ba572db..165f366 100644
--- a/tests/unit/test_evaluator.py
+++ b/tests/unit/test_evaluator.py
@@ -3,7 +3,7 @@
 from featureflags.evaluations.auth_target import Target
 from featureflags.evaluations.constants import (EQUAL_OPERATOR,
                                                 STARTS_WITH_OPERATOR)
-from featureflags.evaluations.enum import FeatureState
+from featureflags.openapi.config.models import FeatureState
 from featureflags.evaluations.evaluator import Evaluator
 from featureflags.lru_cache import LRUCache
 from featureflags.openapi.config.models.clause import Clause