fix a little ..