diff --git a/Dockerfile b/Dockerfile
index 45ee588..54c67c5 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -14,11 +14,11 @@ COPY --from=planner /app/recipe.json recipe.json
 RUN cargo chef cook --release --recipe-path recipe.json
 
 COPY . .
-RUN cargo build --release --bin bootstrap
+RUN cargo build --release --bin bootstrap --features="aws"
 RUN cargo build --release --bin preload
 # The binary will be at target/release/bootstrap (or your binary name)
 RUN cp target/release/bootstrap bootstrap 2>/dev/null || \
-    cp target/release/embedding_service bootstrap
+    cp target/release/serverless_vectorizer bootstrap
 
 RUN cp target/release/preload preload
 
diff --git a/README.md b/README.md
index df7cb99..e51971c 100644
--- a/README.md
+++ b/README.md
@@ -1,6 +1,6 @@
 # Serverless Vectorizer
 
-[![CI](https://github.com/johnnywale/serverless-vectorizer/actions/workflows/ci.yml/badge.svg)](https://github.com/johnnywale/serverless-vectorizer/actions/workflows/ci.yml)
+[![CI](https://github.com/johnnywale/serverless-vectorizer/actions/workflows/test-and-tag.yml/badge.svg)](https://github.com/johnnywale/serverless-vectorizer/actions/workflows/ci.yml)
 [![Release](https://img.shields.io/github/v/release/johnnywale/serverless-vectorizer)](https://github.com/johnnywale/serverless-vectorizer/releases)
 [![License](https://img.shields.io/github/license/johnnywale/serverless-vectorizer)](LICENSE-MIT)
 [![Docker Pulls](https://img.shields.io/docker/pulls/johnnywale/serverless-vectorizer)](https://hub.docker.com/r/johnnywale/serverless-vectorizer)