small fix on speed up