diff --git a/tests/test_util.py b/library/test_util.py
similarity index 100%
rename from tests/test_util.py
rename to library/test_util.py
diff --git a/tests/__init__.py b/tests/__init__.py
deleted file mode 100644
index e69de29b..00000000
diff --git a/tests/library/test_lora_util.py b/tests/library/test_lora_util.py
index df5243ba..60273464 100644
--- a/tests/library/test_lora_util.py
+++ b/tests/library/test_lora_util.py
@@ -1,7 +1,7 @@
 import torch
 import pytest
 from library.lora_util import initialize_pissa
-from ..test_util import generate_synthetic_weights
+from library.test_util import generate_synthetic_weights
 
 
 def test_initialize_pissa_basic():
diff --git a/tests/networks/test_lora_flux.py b/tests/networks/test_lora_flux.py
index 86ccb376..5f141e4f 100644
--- a/tests/networks/test_lora_flux.py
+++ b/tests/networks/test_lora_flux.py
@@ -2,7 +2,7 @@ import pytest
 import torch
 import torch.nn as nn
 from networks.lora_flux import LoRAModule, LoRANetwork, create_network
-from ..test_util import generate_synthetic_weights
+from library.test_util import generate_synthetic_weights
 from unittest.mock import MagicMock