[SPARK-3418] Fixed one more style issue