Update for new key package