update after_each block