Update key after save