diff --git a/src/aeros_equipment/service.py b/src/aeros_equipment/service.py
index 174e8ea..9c6cbb7 100644
--- a/src/aeros_equipment/service.py
+++ b/src/aeros_equipment/service.py
@@ -313,9 +313,16 @@ async def update_equipment_for_simulation(*, db_session: DbSession, project_name
                 reliabiility = reliability_data.get(eq["equipmentName"], {})
 
                 if custom_input and eq["equipmentName"] in custom_input:
-                    eq["cmDisP1"] = reliabiility.get("cmDisP1", 0)
+                    custom_param = custom_input[eq["equipmentName"]]
+
+                    if not custom_param["mttr"]:
+                        continue
+                    if not custom_param['failure_rate']:
+                        continue
+
+                    eq["cmDisP1"] = custom_param["mttr"]
                     eq["relDisType"] = "Fixed"
-                    eq["relDisP1"] = float(custom_input[eq["equipmentName"]])
+                    eq["relDisP1"] = float(custom_param["failure_rate"])
                     eq["relDisP2"] = 0
 
                     reqNodeInputs.append(eq)