diff --git a/tests/unit/tf/examples/test_usecase_retrieval_with_hpo.py b/tests/unit/tf/examples/test_usecase_retrieval_with_hpo.py
index f94c6333d9..e7fa61b1a5 100644
--- a/tests/unit/tf/examples/test_usecase_retrieval_with_hpo.py
+++ b/tests/unit/tf/examples/test_usecase_retrieval_with_hpo.py
@@ -1,7 +1,10 @@
+import pytest
 from testbook import testbook
 
 from tests.conftest import REPO_ROOT
 
+optuna = pytest.importorskip("optuna")
+
 
 @testbook(
     REPO_ROOT / "examples/usecases/retrieval-with-hyperparameter-optimization.ipynb", execute=False