Minor fix in Tests
Minor fix in Tests

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