Invitation for either email or source, and assign group

[serene turtle]

Hi all, I'm trying to make a flow which lets someone accept an invite, then register with either an email and password, or a source. Then it should create the user with a specific group.

If I try enroll with a source, it creates the user but then redirects to the dashboard without assigning the group. If I try enroll with email and password, I get the error in the image below.

Here's the flow

Would be grateful if anyone could help me figure out what's wrong