fix a mini bug.