diff --git a/src/js/game/hud/parts/puzzle_editor_settings.js b/src/js/game/hud/parts/puzzle_editor_settings.js
index 13564da4..bd738198 100644
--- a/src/js/game/hud/parts/puzzle_editor_settings.js
+++ b/src/js/game/hud/parts/puzzle_editor_settings.js
@@ -149,8 +149,9 @@ export class HUDPuzzleEditorSettings extends BaseHUDPart {
                         assertAlways(false, "Failed to re-place building in trim");
                     }
 
-                    if (building.components.ConstantSignal) {
-                        result.components.ConstantSignal.signal = building.components.ConstantSignal.signal;
+                    for (const key in building.components) {
+                        /** @type {import("../../../core/global_registries").Component} */ (building
+                            .components[key]).copyAdditionalStateTo(result.components[key]);
                     }
                 }
             });