diff --git a/scripts/model_builder/plot_data.py b/scripts/model_builder/plot_data.py
index 7016558..8acc1ff 100644
--- a/scripts/model_builder/plot_data.py
+++ b/scripts/model_builder/plot_data.py
@@ -5,10 +5,9 @@ import torch
 from sklearn.metrics import classification_report
 
 os.environ["NNPACK"] = "0"
-from cm_plotter import plot_confusion_matrix
 from matplotlib import pyplot as plt
-from scripts.training.mobilenetv3 import RFClassifier, mobilenetv3
-from scripts.training.modulation_dataset import ModulationH5Dataset
+from mobilenetv3 import RFClassifier, mobilenetv3
+from modulation_dataset import ModulationH5Dataset
 
 from helpers.app_settings import get_app_settings