diff --git a/tests/agents/models/hf.py b/tests/agents/models/hf.py
index 68648904..292b94ba 100644
--- a/tests/agents/models/hf.py
+++ b/tests/agents/models/hf.py
@@ -2,7 +2,7 @@ import pytest
 import torch
 from unittest.mock import Mock, MagicMock
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
-from your_module import HuggingFaceLLM  # replace with actual import
+from swarms.agents.models.hf import HuggingFaceLLM  # replace with actual import
 
 
 @pytest.fixture