minor modification

Signed-off-by: Fei Guan <feiguan@172-16-189-96.dynapool.nyu.edu>