diff --git a/configuration/automations/3d-printer.yaml b/configuration/automations/3d-printer.yaml
index f35bbc0..bd7a3d7 100644
--- a/configuration/automations/3d-printer.yaml
+++ b/configuration/automations/3d-printer.yaml
@@ -7,9 +7,14 @@
       below: 30
       for: '00:30:00'
   condition:
-    - condition: state
-      entity_id: sensor.prusa_mk39
-      state: idle
+    - condition: or
+      conditions:
+        - condition: state
+          entity_id: sensor.prusa_mk39
+          state: finished
+        - condition: state
+          entity_id: sensor.prusa_mk39
+          state: idle
   action:
     - service: switch.turn_off
       target: