diff --git a/.gitignore b/.gitignore
index 8524ead..d7016a2 100644
--- a/.gitignore
+++ b/.gitignore
@@ -17,3 +17,4 @@ inv/
 scripts/
 docx/
 agentic_security.toml
+/venv
\ No newline at end of file
diff --git a/agentic_security/probe_actor/refusal.py b/agentic_security/probe_actor/refusal.py
index a135423..2d5b077 100644
--- a/agentic_security/probe_actor/refusal.py
+++ b/agentic_security/probe_actor/refusal.py
@@ -1,6 +1,6 @@
 from abc import ABC, abstractmethod
 
-from agentic_security.refusal_classifier import RefusalClassifier
+from agentic_security.refusal_classifier.model import RefusalClassifier
 
 classifier = RefusalClassifier()
 classifier.load_model()
diff --git a/agentic_security/refusal_classifier/test_model.py b/tests/refusal_classifier/test_model.py
similarity index 95%
rename from agentic_security/refusal_classifier/test_model.py
rename to tests/refusal_classifier/test_model.py
index b1f017b..1a4677b 100644
--- a/agentic_security/refusal_classifier/test_model.py
+++ b/tests/refusal_classifier/test_model.py
@@ -3,7 +3,7 @@ from unittest.mock import patch
 import pandas as pd
 import pytest
 
-from .model import RefusalClassifier
+from agentic_security.refusal_classifier.model import RefusalClassifier
 
 
 @pytest.fixture