diff --git a/tests/unit/daq/test_aaredb.py b/tests/unit/daq/test_aaredb.py
index ead5aea6..c659060c 100644
--- a/tests/unit/daq/test_aaredb.py
+++ b/tests/unit/daq/test_aaredb.py
@@ -98,7 +98,7 @@ def test_create_manual_sample(mock_sample, mock_api, mock_bl, sample_info):
 def test_send_sample_event(mock_sample, mock_api, mock_bl, sample_info):
     from aareDB import SampleEventType
     wrapper = AareWrapper(bl=mock_bl)
-    wrapper.send_sample_event(sample_info, SampleEventType.MOUNTED, "Test comment")
+    wrapper.send_sample_event(sample_info.db_id, SampleEventType.MOUNTED, "Test comment")
     mock_sample.return_value.create_sample_event.assert_called_once()
 
     # Test None sample
@@ -108,7 +108,7 @@ def test_send_sample_event(mock_sample, mock_api, mock_bl, sample_info):
 
     # Test invalid ID
     sample_info.db_id = -1
-    wrapper.send_sample_event(sample_info, SampleEventType.MOUNTED)
+    wrapper.send_sample_event(sample_info.db_id, SampleEventType.MOUNTED)
     mock_sample.return_value.create_sample_event.assert_not_called()
 
 @patch("aareDB.ApiClient")
@@ -129,7 +129,7 @@ def test_send_sample_event_error(mock_sample, mock_api, mock_bl, sample_info, ca
     caplog.set_level(logging.ERROR)
     wrapper = AareWrapper(bl=mock_bl)
     mock_sample.return_value.create_sample_event.side_effect = Exception("Event Error")
-    wrapper.send_sample_event(sample_info, SampleEventType.MOUNTED)
+    wrapper.send_sample_event(sample_info.db_id, SampleEventType.MOUNTED)
     assert "Error sending sample event" in caplog.text
     assert "MOUNTED" in caplog.text
     assert "Event Error" in caplog.text