Redirect to Registration without loosing url path

We are using UltimateMember to password protect a photo gallery on our site. We are also using Weglot to handle translations. The way it works is it appends a language code to the url after the domain name like: https://siteurl/es/etc, https://siteurl/en/etc., https://siteurl/fr/etc. We have it setup so when a non-logged in user goes to the gallery, UltimateMember redirects to the registration page. When this happens it looses the language code in the url. Is there a way through settings or through code to have it keep the language code after the base url when redirecting? This can not be hardcoded into the redirect path. Thanks