diff --git a/README.md b/README.md
index b677bdb..673b816 100644
--- a/README.md
+++ b/README.md
@@ -1,5 +1,5 @@
 # Masked-AI
-![ci](https://github.com/cado-security/masked-ai/actions/workflows/app-ci.yml/badge.svg?branch=main)
+<!-- ![ci](https://github.com/cado-security/masked-ai/actions/workflows/app-ci.yml/badge.svg?branch=main) -->
 <!-- [![PyPI version](https://badge.fury.io/py/masked-ai.svg)](https://badge.fury.io/py/masked-ai) -->
 
 Masked-AI is a Python SDK and CLI wrappers that enable the usage of public LLM (Language Model) APIs such as OpenAI/GPT4 more securely.
diff --git a/setup.py b/setup.py
index 039b544..7825bbd 100644
--- a/setup.py
+++ b/setup.py
@@ -24,7 +24,7 @@
     author_email='maskedai@cadosecurity.com',
     url='https://github.com/cado-security/masked-ai',
     download_url='https://github.com/cado-security/masked-ai/archive/refs/heads/main.zip',
-    py_modules=['masked_ai'],
+    # py_modules=['masked_ai'],
     install_requires=['nltk'],
     packages=find_packages()
-)
+)
\ No newline at end of file