Fix atomic update code example