diff --git a/pyproject.toml b/pyproject.toml
index 83607539..e44881fa 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -4,7 +4,9 @@ version = "0.1.7"
 description = "Generative AI CodeGen security gateway"
 readme = "README.md"
 authors = []
-
+packages = [
+    { include = "codegate", from = "src" },
+]
 [tool.poetry.dependencies]
 python = ">=3.12,<4.0"
 click = "==8.1.8"
diff --git a/tests/integration/integration_tests.py b/tests/integration/integration_tests.py
index 96ebcdfc..1326d193 100644
--- a/tests/integration/integration_tests.py
+++ b/tests/integration/integration_tests.py
@@ -136,6 +136,8 @@ async def run_test(self, test: dict, test_headers: dict) -> bool:
         streaming = data.get("stream", False)
         provider = test["provider"]
 
+        logger.info(f"Starting test: {test_name}")
+
         response = self.call_codegate(url, test_headers, data, provider)
         if not response:
             logger.error(f"Test {test_name} failed: No response received")