diff --git a/Resources/Prototypes/Recipes/Construction/Graphs/utilities/wallmount_generator.yml b/Resources/Prototypes/Recipes/Construction/Graphs/utilities/wallmount_generator.yml
index b9ca0af5f3..3df8d9c818 100644
--- a/Resources/Prototypes/Recipes/Construction/Graphs/utilities/wallmount_generator.yml
+++ b/Resources/Prototypes/Recipes/Construction/Graphs/utilities/wallmount_generator.yml
@@ -31,6 +31,7 @@
           doAfter: 2
 
     - node: parts
+      entity: BaseGeneratorWallmountFrame
       edges:
       - to: electronics
         steps:
diff --git a/Resources/Prototypes/Recipes/Construction/Graphs/utilities/wallmount_substation.yml b/Resources/Prototypes/Recipes/Construction/Graphs/utilities/wallmount_substation.yml
index ff73e90fd9..78a1a34c6a 100644
--- a/Resources/Prototypes/Recipes/Construction/Graphs/utilities/wallmount_substation.yml
+++ b/Resources/Prototypes/Recipes/Construction/Graphs/utilities/wallmount_substation.yml
@@ -31,6 +31,7 @@
           doAfter: 2
 
     - node: parts
+      entity: BaseSubstationWallFrame
       edges:
       - to: electronics
         steps: