From cae4f8b9342b15562fff93c5d80b37cb38f8af41 Mon Sep 17 00:00:00 2001
From: wyzula-jan <133381102+wyzula-jan@users.noreply.github.com>
Date: Mon, 13 Nov 2023 13:53:08 +0100
Subject: [PATCH] refactor: fix bec_lib imports in tests

---
 bec_widgets/widgets/monitor/monitor.py | 2 +-
 tests/test_bec_monitor.py              | 1 -
 2 files changed, 1 insertion(+), 2 deletions(-)

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):