diff --git a/models/models.py b/models/models.py
index a29d1327..463242c8 100755
--- a/models/models.py
+++ b/models/models.py
@@ -78,7 +78,7 @@ def create_model(opt):
         modelD.initialize(opt)
         flowNet.initialize(opt)        
         if not opt.fp16:
-            modelG, modelD, flownet = wrap_model(opt, modelG, modelD, flowNet)
+            modelG, modelD, flowNet = wrap_model(opt, modelG, modelD, flowNet)
         return [modelG, modelD, flowNet]
     else:
         return modelG