diff --git a/src/utils.py b/src/utils.py
index 597f547..2b3b52e 100755
--- a/src/utils.py
+++ b/src/utils.py
@@ -1882,7 +1882,7 @@ def update_training_dataplot(x_min=None, x_max=None, y_min=None, y_max=None, con
 			losses = gr.LinePlot.update(
 				value = pd.DataFrame(training_state.statistics['loss']),
 				x_lim=x_lim, y_lim=y_lim,
-				x="it", y="value", # x="epoch",
+				x="epoch", y="value", # x="it",
 				title="Loss Metrics", color="type", tooltip=['epoch', 'it', 'value', 'type'],
 				width=500, height=350
 			)
@@ -1890,7 +1890,7 @@ def update_training_dataplot(x_min=None, x_max=None, y_min=None, y_max=None, con
 			lrs = gr.LinePlot.update(
 				value = pd.DataFrame(training_state.statistics['lr']),
 				x_lim=x_lim,
-				x="it", y="value", # x="epoch",
+				x="epoch", y="value", # x="it",
 				title="Learning Rate", color="type", tooltip=['epoch', 'it', 'value', 'type'],
 				width=500, height=350
 			)
@@ -1898,7 +1898,7 @@ def update_training_dataplot(x_min=None, x_max=None, y_min=None, y_max=None, con
 			grad_norms = gr.LinePlot.update(
 				value = pd.DataFrame(training_state.statistics['grad_norm']),
 				x_lim=x_lim,
-				x="it", y="value", # x="epoch",
+				x="epoch", y="value", # x="it",
 				title="Gradient Normals", color="type", tooltip=['epoch', 'it', 'value', 'type'],
 				width=500, height=350
 			)