minor bug fixed

@natashaborzdoun