diff --git a/tests/test_history.py b/tests/test_history.py
index 1f8bef9..f4b5d8c 100644
--- a/tests/test_history.py
+++ b/tests/test_history.py
@@ -4,6 +4,7 @@ import pytest
 import tempfile
 import os
 from pathlib import Path
+from unittest.mock import patch
 
 from shellgenius.history import HistoryStorage, HistoryLearner, HistoryEntry