Can't edit the new customisable touch controls in Survival

The new customisable touch controls cannot be edited in a survival world.
Steps to recreate:-
1. Go to settings, Touch and enable customisable touch controls.
2. Go to a SURVIVAL world.
3. Try to edit the controls.
Observation:- You get the message "these controls cannot be saved"
I tried editing the controls in many ways but nothing worked
I really want this feature in my survival world. This is going to be a game changer. I will be very thankful to you if you fix this bug.
I play minecraft on an Android 12 tablet.