diff --git a/cyclegan/models/test_model.py b/cyclegan/models/test_model.py
index 6a4ff8a..39f85c1 100644
--- a/cyclegan/models/test_model.py
+++ b/cyclegan/models/test_model.py
@@ -1,5 +1,6 @@
 from .base_model import BaseModel
 from . import networks
+from util import attacks
 
 
 class TestModel(BaseModel):