TEZ-321. Fix build break due to changes from YARN-875. (hitesh)