diff --git a/cmd/demo/BUILD.bazel b/cmd/demo/BUILD.bazel
index 5fb6177..b949e95 100644
--- a/cmd/demo/BUILD.bazel
+++ b/cmd/demo/BUILD.bazel
@@ -28,12 +28,7 @@ load("@rules_oci//oci:defs.bzl", "oci_image")
 
 oci_image(
     name = "image",
-    architecture = select({
-        "@platforms//cpu:arm64": "arm64",
-        "@platforms//cpu:x86_64": "amd64",
-    }),
-    base = ":distroless_base",
-    entrypoint = ["/app"],
-    os = "linux",
+    base = "//:distroless_base",
+    entrypoint = ["/demo"],
     tars = [":tar"],
 )