fix a slight bu