Issue with Membership 2 WPML integration

WPML integration add-ons doesn't seem to work after activation. When clicking the language icon to edit notification subject and message, the language stays the same(as edited) even after switching language from WPML. See attachment.

And sometimes the subject is not saved with the new text.