little fix in one function