diff --git a/custom_routes.py b/custom_routes.py
index 18788ce..2dc2d76 100644
--- a/custom_routes.py
+++ b/custom_routes.py
@@ -245,7 +245,7 @@ def apply_random_seed_to_workflow(workflow_api):
                 workflow_api[key]['inputs']['seed'] = randomSeed()
                 logger.info(f"Applied random seed {workflow_api[key]['inputs']['seed']} to {workflow_api[key]['class_type']}")
 
-            if 'noise_seed' in workflow_api[key]:
+            if 'noise_seed' in workflow_api[key]['inputs']:
                 if workflow_api[key]['class_type'] == "RandomNoise":
                     workflow_api[key]['inputs']['noise_seed'] = randomSeed()
                     logger.info(f"Applied random noise_seed {workflow_api[key]['inputs']['noise_seed']} to RandomNoise")