перенеси эту логику в UpdateContext - там правильное место для неё