fix a small  bug