diff --git a/sql/include/dockerfile_content.go b/sql/include/dockerfile_content.go
index 2b7b0cdc0..cbcef92ff 100644
--- a/sql/include/dockerfile_content.go
+++ b/sql/include/dockerfile_content.go
@@ -7,6 +7,8 @@ import "strings"
 var Dockerfile = strings.TrimSpace(`
 FROM python:%s-slim-bullseye
 
+ENV ASTRO_CLI Yes
+
 RUN apt-get update -y && apt-get install -y gcc g++ libpq-dev python-dev libssl-dev libffi-dev
 
 RUN pip install astro-sql-cli==%s