From 1d99f5ad29efdbffa7b761b7a142440aa7f64e58 Mon Sep 17 00:00:00 2001
From: Mick Vleeshouwer <mick@imick.nl>
Date: Mon, 1 Nov 2021 21:18:45 +0100
Subject: [PATCH] Fix tests (2)

---
 pyhoma/models.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/pyhoma/models.py b/pyhoma/models.py
index 9f4ccedb..e03e5d44 100644
--- a/pyhoma/models.py
+++ b/pyhoma/models.py
@@ -5,6 +5,7 @@
 
 import attr
 
+from pyhoma.enum.gateway import GatewaySubType, GatewayType
 from pyhoma.enums import (
     DataType,
     EventName,
@@ -12,8 +13,6 @@
     ExecutionSubType,
     ExecutionType,
     FailureType,
-    GatewaySubType,
-    GatewayType,
     ProductType,
     UpdateBoxStatus,
 )