diff --git a/tests/test_auto_whitelist.py b/tests/test_auto_whitelist.py
index 5ad2c77d..58ee2eff 100644
--- a/tests/test_auto_whitelist.py
+++ b/tests/test_auto_whitelist.py
@@ -1,3 +1,4 @@
+import dataclasses
 import pytest
 from src.log_registry import LogRegistry
 from pathlib import Path
@@ -17,7 +18,7 @@ def test_auto_whitelist_keywords(registry_setup: LogRegistry) -> None:
  reg.register_session(session_id, "logs", start_time)
  
  # Manual override for testing if log files don't exist
- reg.data[session_id]["whitelisted"] = True
+ reg.data[session_id] = dataclasses.replace(reg.data[session_id], whitelisted=True)
  assert reg.is_session_whitelisted(session_id) is True
 
 def test_auto_whitelist_message_count(registry_setup: LogRegistry) -> None: