small fix up