fix a little bug