Fix broken test.
Fix broken test.

eric-haibin-lin/peloton,eric-haibin-lin/peloton,eric-haibin-lin/peloton,eric-haibin-lin/peloton,eric-haibin-lin/peloton,eric-haibin-lin/peloton,eric-haibin-lin/peloton