diff --git a/stargan/solver.py b/stargan/solver.py
index 245ffc6..1e98f63 100644
--- a/stargan/solver.py
+++ b/stargan/solver.py
@@ -569,7 +569,7 @@ class Solver(object):
                 save_image(self.denorm(x_concat.data.cpu()), result_path, nrow=1, padding=0)
                 print('Saved real and fake images into {}...'.format(result_path))
 
-    def test_attack(self):
+    def test_attack_output(self):
         """Translate images using StarGAN trained on a single dataset."""
 
         layer_dict = {0: 2, 1: 5, 2: 8, 3: 9, 4: 10, 5: 11, 6: 12, 7: 13, 8: 14, 9: 17, 10: 20, 11: None}
@@ -654,7 +654,7 @@ class Solver(object):
         print('{} images. L1 error: {}. L2 error: {}. L0 error: {}. L_-inf error: {}. Perceptual error: {}.'.format(n_samples, 
         l1_error / n_samples, l2_error / n_samples, l0_error / n_samples, min_dist / n_samples, perceptual_error / n_samples))
 
-    def test_attack_feats(self):
+    def test_attack(self):
         """Translate images using StarGAN trained on a single dataset."""
 
         layer_dict = {0: 2, 1: 5, 2: 8, 3: 9, 4: 10, 5: 11, 6: 12, 7: 13, 8: 14, 9: 17, 10: 20, 11: None}