Fix dry mode when using partial updates