diff --git a/infrastructure/aws/cdk/app.py b/infrastructure/aws/cdk/app.py
index 0a62609..592cfd9 100644
--- a/infrastructure/aws/cdk/app.py
+++ b/infrastructure/aws/cdk/app.py
@@ -116,6 +116,9 @@ def __init__(
                 path=os.path.abspath(context_dir),
                 file="infrastructure/aws/lambda/Dockerfile",
                 platform="linux/amd64",
+                build_args={
+                    "PYTHON_VERSION": runtime.to_string().replace("python", ""),
+                },
             ),
             handler="handler.handler",
             memory_size=memory,