fix a little stuff