just small fix up