From dad6b44e8d4aaf29472810531752e8be663a6afd Mon Sep 17 00:00:00 2001
From: Ben van Basten <ben.vanbasten@nelen-schuurmans.nl>
Date: Tue, 26 Nov 2024 13:57:25 +0100
Subject: [PATCH] fixed test

---
 threedigrid/admin/structure_controls/models.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/threedigrid/admin/structure_controls/models.py b/threedigrid/admin/structure_controls/models.py
index 8cdc82f..4799cdc 100644
--- a/threedigrid/admin/structure_controls/models.py
+++ b/threedigrid/admin/structure_controls/models.py
@@ -1,4 +1,4 @@
-from enum import Enum, StrEnum
+from enum import Enum
 from typing import List
 
 
@@ -8,7 +8,7 @@ class StructureControlTypes(Enum):
     timed_control = "timed_control"
 
 
-class StructureControlSourceTypes(StrEnum):
+class StructureControlSourceTypes(str, Enum):
     LINES = "lines"
     NODES = "nodes"
     PUMPS = "pumps"