[HOTFIX] Fix compilation error for Yarn 2.0.*-alpha

This was reported in https://issues.apache.org/jira/browse/SPARK-3445.