Update test because of YARN-903