diff --git a/Espresso_Water_Meter/code.py b/Espresso_Water_Meter/code.py
index fba430e16..5b7c976d3 100644
--- a/Espresso_Water_Meter/code.py
+++ b/Espresso_Water_Meter/code.py
@@ -125,8 +125,8 @@ def set_pixel_color(distance):
         # pylint: disable=invalid-name
         avg_distance = 22
     print(f"Average distance: {avg_distance:.1f} cm")
-    # Set color based on average distance
-    set_pixel_color(avg_distance)
+    # Set color based on average distance - Moved down after connection test
+    # set_pixel_color(avg_distance)
 
     # Check battery status
     battery_voltage = battery_monitor.cell_voltage
@@ -151,6 +151,9 @@ def set_pixel_color(distance):
         time.sleep(10)
         microcontroller.reset()
 
+    # Set color based on average distance
+    set_pixel_color(avg_distance)
+
     # Create a socket pool
     pool = socketpool.SocketPool(wifi.radio)
     requests = adafruit_requests.Session(pool, ssl.create_default_context())