diff --git a/cyclegan/test.py b/cyclegan/test.py
index 56a5dbd..1caabeb 100644
--- a/cyclegan/test.py
+++ b/cyclegan/test.py
@@ -32,7 +32,7 @@ from data import create_dataset
 from models import create_model
 from util.visualizer import save_images
 from util import html
-
+import torch
 
 if __name__ == '__main__':
     opt = TestOptions().parse()  # get test options