fix a litte bug