a little problem fix up