updating logics
minor fix