diff --git a/bec_widgets/widgets/monitor/monitor.py b/bec_widgets/widgets/monitor/monitor.py
index adc26836..b53bc400 100644
--- a/bec_widgets/widgets/monitor/monitor.py
+++ b/bec_widgets/widgets/monitor/monitor.py
@@ -4,7 +4,7 @@ import time
 import pyqtgraph as pg
 from pydantic import ValidationError
 
-from bec_lib.core import MessageEndpoints
+from bec_lib import MessageEndpoints
 from PyQt5 import QtCore
 from PyQt5.QtCore import pyqtSignal, pyqtSlot
 from PyQt5.QtWidgets import QApplication, QTableWidgetItem, QWidget, QMessageBox
diff --git a/tests/test_bec_monitor.py b/tests/test_bec_monitor.py
index af6491bd..6e9f9b00 100644
--- a/tests/test_bec_monitor.py
+++ b/tests/test_bec_monitor.py
@@ -5,7 +5,6 @@ import pytest
 from unittest.mock import MagicMock
 
 from bec_widgets.widgets import BECMonitor
-from bec_widgets.validation import MonitorConfigValidator
 
 
 def load_test_config(config_name):