diff --git a/Dockerfile b/Dockerfile
index 9c2ec97..65d8797 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -39,8 +39,8 @@ ARG index_advisor_release=0.2.0
 ARG supautils_release=2.2.0
 ARG wal_g_release=2.0.1
 
-#FROM nvcr.io/nvidia/tritonserver:24.04-py3-min as base
-FROM ubuntu:focal as base
+FROM nvcr.io/nvidia/tritonserver:24.04-py3-min as base
+#FROM ubuntu:focal as base
 RUN apt update -y && apt install -y \
     curl \
     gnupg \