diff --git a/SQLiteWriter.py b/SQLiteWriter.py
index 7b074c2..7c6d672 100644
--- a/SQLiteWriter.py
+++ b/SQLiteWriter.py
@@ -1,11 +1,16 @@
+from __future__ import annotations
+
 import os
 import sqlite3
+from typing import TYPE_CHECKING
 
 import pandas as pd
 import pyomo.core.base.set
 import pyomo.environ as pyo
 from InOutModule.printer import Printer
-from LEGO.LEGO import LEGO
+
+if TYPE_CHECKING:
+    from LEGO.LEGO import LEGO
 
 printer = Printer.getInstance()