diff --git a/source/models.py b/source/models.py
index 3c70bf8..762e277 100644
--- a/source/models.py
+++ b/source/models.py
@@ -155,7 +155,7 @@ def fit(
                     # add to history and plot
                     loss_history.append(train_loss)
                     acc_history.append(train_acc)
-                    utils.plot_loss_and_accuracy(loss_history, acc_history, data_directory="cbow")
+                    utils.plot_loss_and_accuracy(loss_history, acc_history, data_directory="glove")
                     # update information with current values
                     utils.set_model_progressbar_prefix(progressbar, train_loss, best_loss, train_acc, best_acc)
             # check for early stopping