diff --git a/climax/core/utils.py b/climax/core/utils.py
index 485e1a80ff679e6288329240ecda740b84bbdf6d..4081d02ea3afa80399dcc0ca8f5550523a67481b 100644
--- a/climax/core/utils.py
+++ b/climax/core/utils.py
@@ -179,7 +179,8 @@ def plot_loss(state_file, figsize=(10, 10), step=5):
     yl_max, yl_min = (ceil_decimal(max_loss, decimal=1),
                       floor_decimal(min_loss, decimal=1))
     axes[0].set_ylim(yl_min, yl_max)
-    axes[0].set_yticks(np.arange(yl_min, yl_max + 0.2, 0.2), fontsize=14)
+    axes[0].set_yticks(np.arange(yl_min, yl_max + 0.2, 0.2))
+    axes[0].set_yticklabels(np.arange(yl_min, yl_max + 0.2, 0.2), fontsize=14)
 
     # compute early stopping point
     if loss['valid_loss'].any():