Update some logic