diff --git a/tools/mig-test/misc/patch_flink_conf.rb b/tools/mig-test/misc/patch_flink_conf.rb
index fe60301882..6241a52c34 100644
--- a/tools/mig-test/misc/patch_flink_conf.rb
+++ b/tools/mig-test/misc/patch_flink_conf.rb
@@ -26,7 +26,11 @@
 execution.checkpointing.interval: 300
 EXTRACONF
 
-File.write("#{FLINK_HOME}/conf/flink-conf.yaml", EXTRA_CONF, mode: 'a+')
+if File.file?("#{FLINK_HOME}/conf/flink-conf.yaml")
+  File.write("#{FLINK_HOME}/conf/flink-conf.yaml", EXTRA_CONF, mode: 'a+')
+else
+  File.write("#{FLINK_HOME}/conf/config.yaml", EXTRA_CONF, mode: 'a+')
+end
 
 # MySQL connector is not provided
 `wget https://repo1.maven.org/maven2/mysql/mysql-connector-java/8.0.27/mysql-connector-java-8.0.27.jar -O #{FLINK_HOME}/lib/mysql-connector-java-8.0.27.jar`
\ No newline at end of file