diff --git a/README.md b/README.md
index 91dc6b08..7aedfa84 100644
--- a/README.md
+++ b/README.md
@@ -10,9 +10,8 @@
   <img src="https://img.shields.io/badge/License-MIT-yellow.svg" alt="License">
 </a>
   <img src="https://img.shields.io/github/last-commit/ArcadeAI/arcade-ai" alt="GitHub last commit">
-</a>
-<a href="https://github.com/arcadeai/arcade-ai/actions/workflow/on-release-main.yml">
-<img src="https://img.shields.io/github/actions/workflow/status/arcadeai/arcade-ai/check-toolkits.yml" alt="GitHub Actions Status">
+<a href="https://github.com/arcadeai/arcade-ai/actions?query=branch%3Amain">
+<img src="https://img.shields.io/github/actions/workflow/status/arcadeai/arcade-ai/main.yml?branch=main" alt="GitHub Actions Status">
 </a>
 <a href="https://img.shields.io/pypi/pyversions/arcade-ai">
   <img src="https://img.shields.io/pypi/pyversions/arcade-ai" alt="Python Version">
diff --git a/docker/Dockerfile b/docker/Dockerfile
index 7931679b..75abe0b3 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -33,9 +33,8 @@ EXPOSE $PORT
 # List wheel files for debugging purposes
 RUN ls -la /app/dist/
 
-# Install the core Arcade packages (each has its own version)
+# Install the CLI and worker package
 RUN python -m pip install \
-    /app/dist/arcade_core-*.whl \
     /app/dist/arcade_serve-*.whl \
     /app/dist/arcade_ai-*.whl
 
diff --git a/libs/arcade-cli/arcade_cli/main.py b/libs/arcade-cli/arcade_cli/main.py
index 3ea3be8f..a2e37119 100644
--- a/libs/arcade-cli/arcade_cli/main.py
+++ b/libs/arcade-cli/arcade_cli/main.py
@@ -505,7 +505,7 @@ def serve(
     require_dependency(
         package_name="arcade_serve",
         command_name="serve",
-        install_command=r"pip install 'arcade-ai\[serve]'",
+        install_command=r"pip install 'arcade-serve'",
     )
 
     from arcade_cli.serve import serve_default_worker
@@ -558,7 +558,7 @@ def workerup(
     require_dependency(
         package_name="arcade_serve",
         command_name="worker",
-        install_command=r"pip install 'arcade-ai\[worker]'",
+        install_command=r"pip install 'arcade-serve'",
     )
 
     from arcade_cli.serve import serve_default_worker
diff --git a/libs/arcade-cli/arcade_cli/templates/{{ toolkit_name }}/Makefile b/libs/arcade-cli/arcade_cli/templates/{{ toolkit_name }}/Makefile
index e1849fe5..7e2c686e 100644
--- a/libs/arcade-cli/arcade_cli/templates/{{ toolkit_name }}/Makefile	
+++ b/libs/arcade-cli/arcade_cli/templates/{{ toolkit_name }}/Makefile	
@@ -33,6 +33,13 @@ test: ## Test the code with pytest
 	@echo "🚀 Testing code: Running pytest"
 	@uv run pytest -W ignore -v --cov --cov-config=pyproject.toml --cov-report=xml
 
+.PHONY: coverage
+coverage: ## Generate coverage report
+	@echo "coverage report"
+	coverage report
+	@echo "Generating coverage report"
+	coverage html
+
 .PHONY: bump-version
 bump-version: ## Bump the version in the pyproject.toml file by a patch version
 	@echo "🚀 Bumping version in pyproject.toml"