diff --git a/Dockerfile b/Dockerfile
index d5694ce..5394067 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -10,8 +10,7 @@
 # Make sure RUBY_VERSION matches the Ruby version in .ruby-version
 # ARG RUBY_VERSION=3.3.5
 # FROM docker.io/library/ruby:$RUBY_VERSION-slim AS base
-ARG BASE_IMAGE=docker.io/songtianlun/today_ai_weather:base_main
-FROM ${BASE_IMAGE} AS base
+FROM docker.io/songtianlun/today_ai_weather:base_main AS base
 
 # Rails app lives here
 WORKDIR /rails
@@ -27,8 +26,7 @@ ENV RAILS_ENV="production" \
 
 # Throw-away build stage to reduce size of final image
 # FROM base AS build
-ARG DEVEl_IMAGE=docker.io/songtianlun/today_ai_weather:devel_main
-FROM ${DEVEL_IMAGE} AS build
+FROM docker.io/songtianlun/today_ai_weather:devel_main AS build
 
 # Install packages needed to build gems and node modules
 # Install JavaScript dependencies