diff --git a/.github/workflows/publish.yaml b/.github/workflows/publish.yaml
index 214e5f8..c9a4410 100644
--- a/.github/workflows/publish.yaml
+++ b/.github/workflows/publish.yaml
@@ -4,7 +4,6 @@ on:
   push:
     paths:
       - '**.py'
-      - '!**/tests/**'
   workflow_dispatch: ~
 
 permissions:
diff --git a/tests/example.py b/tests/example.py
index 28ef156..bbff05e 100644
--- a/tests/example.py
+++ b/tests/example.py
@@ -3,7 +3,7 @@
 import openai
 import asyncio
 from openai_streaming import process_response
-from typing import Generator, AsyncGenerator
+from typing import AsyncGenerator
 from openai_streaming import openai_streaming_function
 
 # Initialize API key
diff --git a/tests/test_with_functions.py b/tests/test_with_functions.py
index 2f382e5..791dd15 100644
--- a/tests/test_with_functions.py
+++ b/tests/test_with_functions.py
@@ -5,10 +5,10 @@
 import openai
 from unittest.mock import patch
 from openai_streaming import process_response, openai_streaming_function
-from typing import Generator, AsyncGenerator
+from typing import AsyncGenerator
 
 
-async def content_handler(content: Generator[str, None, None]):
+async def content_handler(content: AsyncGenerator[str, None]):
     async for token in content:
         print(token, end="")