A large emphasis of swimming and water safety in the early stages (bubs, Levels 1 – 3) is not just the confidence to try, and the body positioning and stroke, but the ability to recover and get to safety.
You will notice that almost all the time after an entry and “swim” the learners are either taught to roll onto their backs, or taught to propel towards the edge of a pool or to something that floats. And sometimes even a combination of all these things.