1 files changed, 1 insertions, 1 deletions
diff --git a/models/models.py b/models/models.py
index efcd898..d5bb9d8 100644
--- a/models/models.py
+++ b/models/models.py
@@ -13,7 +13,7 @@ def create_model(opt):
     elif opt.model == 'test':
         assert(opt.dataset_mode == 'single')
         from .test_model import TestModel
-        model = OneDirectionTestModel()
+        model = TestModel()
     else:
         raise ValueError("Model [%s] not recognized." % opt.model)
     model.initialize(opt)