fix a little problem