diff --git a/README.md b/README.md
index 15ed09d4e..2acd0c5a4 100644
--- a/README.md
+++ b/README.md
@@ -344,7 +344,7 @@ for config_file in ${config_files}; do
 	dir=$(dirname "$config_file")
 	yaml_files=$(find "$dir" -name "*.yaml")
 	for yaml_file in ${yaml_files}; do
-		thrift_file=$(yq -r '.. | .thriftFile? | select(strings | endswith(".thrift"))' "$yaml_file")
+		thrift_file=$(yq -r '.. | .idlFile? | select(strings | endswith(".thrift"))' "$yaml_file")
 		[[ -z ${thrift_file} ]] && continue
 		[[ ${THRIFTRW_SRCS} == *${thrift_file}* ]] && continue
         THRIFTRW_SRCS+=" $CONFIG_DIR/idl/$thrift_file"
diff --git a/codegen/runner/pre-steps.sh b/codegen/runner/pre-steps.sh
index 35e159c96..7668bc33c 100644
--- a/codegen/runner/pre-steps.sh
+++ b/codegen/runner/pre-steps.sh
@@ -144,6 +144,7 @@ for config_file in ${config_files}; do
 		fi
 
 		thrift_file=$(${processor} -r '.. | .thriftFile? | select(strings | endswith(".thrift"))' "$yaml_file")
+		thrift_file+=$(${processor} -r '.. | .idlFile? | select(strings | endswith(".thrift"))' "$yaml_file")
 
 		# process .proto files
 		proto_file=$(${processor} -r '.. | .idlFile? | select(strings | endswith(".proto"))' "$yaml_file")