fix litte bugs