Take a look how Shopware handles this when admin user accepts customer group request / change.
I think you need to use the SalesChannelContextRestorer. So:
$this->customerRepository->update($updateData, $context);
$this->restorer->restoreByCustomer($customer->getId(), $context);