Divi Visual Builder doesn't save a page while using custom code

Hi
I am using the following custom code to redirect a logged in user (with editing rights) after editing a page.

The redirect is to the MemberPress account page. The code works as the user is redirected to the MemberPress account page and not the default WP account page. But when the user tries to save an edit on the page, then he gets the standard 'can't save' message.
If I disable the code snippet, then the save is successful.
I have validated the code and there are no syntax errors.
So, obviously this code is the issue. I looked at the browser console and found an error, but this looks more like a generic issue than something related to this code snippet. All screenshots attached.
It will be great if someone can help with a solution.