Updated to support new update.Key type