This seems to work with a single website but not with multi-websites.

website1.com -> website1.com/website1/
website2.com -> website2.com/website2/

With your module everything redirects to website1.com

it should be

website1.com -> website1.com
website2.com -> website2.com