small fix cont