diff --git a/.templates/90-config_yaml.sh b/.templates/90-config_yaml.sh
index 1f0e73a19..68882274c 100644
--- a/.templates/90-config_yaml.sh
+++ b/.templates/90-config_yaml.sh
@@ -58,7 +58,7 @@ else
 fi
 
 # Check if = instead of :
-if [[ "$(grep -c "=" "config.yaml")" -gt 2 ]]; then
+if [[ "$(grep -c "=" "$CONFIGSOURCE")" -gt 2 ]]; then
     bashio::log.warning 'Are you sure you did not use "KEY=VALUE" ? yaml nomenclature requires "KEY:VALUE"'
 fi