DEV: small updates on WM atlas creation