Should probably put this in an `ensure` block?