diff --git a/arch/arm/mach-rockchip/fit_nodes.sh b/arch/arm/mach-rockchip/fit_nodes.sh
index 9639a06e1c..ceee0ccc20 100755
--- a/arch/arm/mach-rockchip/fit_nodes.sh
+++ b/arch/arm/mach-rockchip/fit_nodes.sh
@@ -180,7 +180,7 @@ function gen_bl32_node()
 		fi
 	fi
 
-	TEE="tee.bin"
+	TEE="${TEE:=tee.bin}"
 	echo "		optee {
 			description = \"OP-TEE\";
 			data = /incbin/(\"${TEE}${SUFFIX}\");