As a workaround, instead of using the Replace WAN Edge feature add the replacement device using the claim code. This ensures that the device is correctly mapped to the appropriate SKU model.
Note: Issue is being investigated internally and will be fixed soon ( article will be updated once fixed )