small refact.