diff --git a/tests/test_models.py b/tests/test_models.py
index b43e00f..d951f44 100644
--- a/tests/test_models.py
+++ b/tests/test_models.py
@@ -90,14 +90,14 @@ def test_normalization_check():
             # so here the first 2 pixels are set to the limits
             test_x[0][0][0] = ra[0]  
             test_x[0][0][1] = ra[1]
-            xrv.models.warning_log = {}
+            xrv.utils.warning_log = {}
             model(test_x)
             assert xrv.utils.warning_log['norm_correct'] == False, ra
             
         for ra in correct_ranges:
             test_x = torch.zeros([1,1,224,224])
             test_x.uniform_(ra[0], ra[1])
-            xrv.models.warning_log = {}
+            xrv.utils.warning_log = {}
             model(test_x)
             assert xrv.utils.warning_log['norm_correct'] == True, ra