small modification

Signed-off-by: ezheng <ezheng@cs.unc.edu>