diff --git a/release_image/main.go b/release_image/main.go
index b5ec73b02..0178ae7cc 100644
--- a/release_image/main.go
+++ b/release_image/main.go
@@ -18,7 +18,7 @@ import (
 )
 
 // next tag to use
-const TAG = "v0.1.6-test7"
+const TAG = "v0.1.6"
 
 // current node gets latest tag...
 const CURRENT_NODE_VERSION = 18
@@ -32,18 +32,18 @@ var NODE_VERSIONS = []int{
 	// 20,
 }
 
-const AUTO_SCHEMA_VERSION = "0.0.32"
-const TSENT_VERSION = "v0.1.6-test4"
+const AUTO_SCHEMA_VERSION = "0.0.30"
+const TSENT_VERSION = "v0.1.6"
 
 var SUFFIXES = []string{
 	"dev",
-	// "slim",
+	"slim",
 }
 
 // can change platforms here to test locally
 var PLATFORMS = []string{
 	"linux/amd64",
-	// "linux/arm64",
+	"linux/arm64",
 }
 
 func do(version int) error {
diff --git a/ts/Dockerfile.tmpl b/ts/Dockerfile.tmpl
index d8e77eaeb..ceccc6ce6 100644
--- a/ts/Dockerfile.tmpl
+++ b/ts/Dockerfile.tmpl
@@ -25,8 +25,8 @@ RUN python -m venv /opt/venv
 # Make sure we use the virtualenv
 ENV PATH /opt/venv/bin:$PATH
 
-# RUN python3 -m pip install auto_schema==$AUTO_SCHEMA_VERSION
-RUN python3 -m pip install --index-url https://test.pypi.org/simple/ --extra-index-url https://pypi.org/simple/ auto_schema_test==$AUTO_SCHEMA_VERSION
+RUN python3 -m pip install auto_schema==$AUTO_SCHEMA_VERSION
+# RUN python3 -m pip install --index-url https://test.pypi.org/simple/ --extra-index-url https://pypi.org/simple/ auto_schema_test==$AUTO_SCHEMA_VERSION
 
 # final image needs to be python-based to have auto-schema work
 FROM python:3.11.0-slim AS final-image
@@ -71,7 +71,7 @@ RUN npm install --save-dev tsconfig-paths@4.2.0 @swc-node/register@1.6.6
 
 COPY --from=golang-image /go/bin/tsent /go/bin/tsent
 COPY --from=python-image /opt/venv /opt/venv
-RUN alias auto_schema=auto_schema_test
+# RUN alias auto_schema=auto_schema_test
 COPY --from=golang-image /go/pkg/mod/github.com/lolopinto/ent@$GOLANG_VERSION /go/pkg/mod/github.com/lolopinto/ent@$GOLANG_VERSION
 
 CMD ["node"]