diff --git a/examples/aws/poc/dsf_deployment/agent_sources.tf b/examples/aws/poc/dsf_deployment/agent_sources.tf
index aca266bd0..06399eb60 100644
--- a/examples/aws/poc/dsf_deployment/agent_sources.tf
+++ b/examples/aws/poc/dsf_deployment/agent_sources.tf
@@ -1,5 +1,5 @@
 locals {
-  db_types_for_agent = local.create_agent_gw_cluster > 0 ? var.simulation_db_types_for_agent : []
+  db_types_for_agent = local.agent_gw_count > 0 ? var.simulation_db_types_for_agent : []
 }
 
 module "db_with_agent" {