diff --git a/stargan/solver.py b/stargan/solver.py
index d97cab0..45e07bc 100644
--- a/stargan/solver.py
+++ b/stargan/solver.py
@@ -609,12 +609,13 @@ class Solver(object):
             # Translate images.
             x_fake_list = [x_real]
 
-            x_advs = []
+            # x_advs = []
             if i == 0:
-                for idx, c_trg in enumerate(c_trg_list):
-                    x_adv, perturb = pgd_attack.perturb(x_real, black, c_trg)
-                    x_advs.append((x_adv, perturb))
-                    break
+                x_adv, perturb = pgd_attack.perturb(x_real, black, c_trg_list[0])
+                # for idx, c_trg in enumerate(c_trg_list):
+                #     x_adv, perturb = pgd_attack.perturb(x_real, black, c_trg)
+                #     x_advs.append((x_adv, perturb))
+                #     break
             
             for idx, c_trg in enumerate(c_trg_list):
                 with torch.no_grad():