From 197a13ccb923863dd2babd04789c888eccde8b84 Mon Sep 17 00:00:00 2001
From: MaxMLang <langmaxmelchior@gmail.com>
Date: Sun, 7 Apr 2024 04:20:50 +0100
Subject: [PATCH] Testing fixes

---
 tests/test_pytector.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_pytector.py b/tests/test_pytector.py
index f3adc43..0ba8074 100644
--- a/tests/test_pytector.py
+++ b/tests/test_pytector.py
@@ -1,5 +1,5 @@
 import unittest
-from src import pytector
+from pytector import PromptInjectionDetector
 
 class TestPromptInjectionDetector(unittest.TestCase):