diff --git a/demos/weather_forecast/docker-compose-logfire.yaml b/demos/weather_forecast/docker-compose-logfire.yaml
index 92371d51..6f747cd0 100644
--- a/demos/weather_forecast/docker-compose-logfire.yaml
+++ b/demos/weather_forecast/docker-compose-logfire.yaml
@@ -1,7 +1,7 @@
 services:
   weather_forecast_service:
     build:
-      context: ./weather_forecast_service
+      context: ./
     environment:
       - OLTP_HOST=http://otel-collector:4317
     extra_hosts:
diff --git a/demos/weather_forecast/docker-compose-signoz.yaml b/demos/weather_forecast/docker-compose-signoz.yaml
index e9e2e536..392debfb 100644
--- a/demos/weather_forecast/docker-compose-signoz.yaml
+++ b/demos/weather_forecast/docker-compose-signoz.yaml
@@ -4,7 +4,7 @@ include:
 services:
   weather_forecast_service:
     build:
-      context: ./weather_forecast_service
+      context: .
     environment:
       - OLTP_HOST=http://otel-collector:4317
     extra_hosts:
diff --git a/demos/weather_forecast/run_demo.sh b/demos/weather_forecast/run_demo.sh
index a129e2c6..e764c9fa 100644
--- a/demos/weather_forecast/run_demo.sh
+++ b/demos/weather_forecast/run_demo.sh
@@ -56,7 +56,7 @@ start_demo() {
 
   load_env
 
-  if [ "$1" == "logfire" ] && [ -z "$LOGFIRE_API_KEY"]; then
+  if [ "$1" == "logfire" ] && [ -z "$LOGFIRE_API_KEY" ]; then
     echo "Error: LOGFIRE_API_KEY environment variable is required for Logfire."
     exit 1
   fi