Swimming in shallow water and small sources

In Bedrock, you can't initiate swimming in water that is either one block deep or when the water source is smaller that one block such as water placed on stairs and slabs. This can be frusturating when trying to swim and move fast in the game. I hope this can be added as it is available in the Legacy Console Editions. The overall water mechanics of the Legacy Console Editions are far more polished and intuitive and I hope to see them in Bedrock, especially the main issue mentioned above.