Touch control: Customisation to the Sneak button gets reset when you go inside scaffolding or water

With touch control and "Swap Jump and Sneak" enabled, customisation applied to the Sneak button gets reset when you go inside scaffolding blocks or water. The button never reverts to the customised position even after you leave scaffolding or water.

Steps to reproduce:

1. Open "Settings"
2. Select the "Touch" tab
3. Select "Change Control Mode"
4. Select "D-Pad & tap to interact"
5. Enable "Swap Jump and Sneak"
6. Enable "Customise Controls" and open the customisation GUI
7. Place the Sneak button somewhere you feel comfortable
8. Tap "Done" and resume the game
9. Go inside scaffolding or water
10. Leave scaffolding or water

What I observe:
The Sneak button goes back to the original position, presumably to prevent "Up" and "Down" buttons from going out of the screen?

What I expect:
I expect either

1. The Sneak button does not move, and two additional "Up" and "Down" buttons are shown right above and below the customised Sneak button,
2. Or the game allows us to reposition each of "Up", "Sneak", and "Down" buttons individually. Currently we can customise the position of "Sneak" and "Down" buttons but the "Up" button is immovable.