fixed one minor bug