fix a little issue