Or perhaps `.update`, and put the line below inside an `else:`?