fixed a segment fault issue

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