diff --git a/stargan/solver.py b/stargan/solver.py
index f02a484..e7fdaf3 100644
--- a/stargan/solver.py
+++ b/stargan/solver.py
@@ -634,8 +634,8 @@ class Solver(object):
                     # gen, gen_feats = self.G(x_adv, c_trg)
 
                     # Add to lists
-                    x_fake_list.append(preproc_x)
-                    # x_fake_list.append(x_adv)
+                    # x_fake_list.append(preproc_x)
+                    x_fake_list.append(x_adv)
                     # x_fake_list.append(perturb)
                     x_fake_list.append(gen)