little fix

Signed-off-by: Hongyi Jia <Hongyi.Jia.GR@dartmouth.edu>