Could it be handled in a more graceful way by showing a banner notification in the admin instead of die()’ing? I’ll write the patch if you’ll consider it.
Additionally, when registering this way the user is sent an activation email. The email links to wp-activate.php which does not load all plugins so I cannot hook into it on that end either.