We’re not using OneNote (yet!), but we are using Word & OneDrive on a lot of pupil devices. Whilst initially this was through the OneDrive for Business app and a bit buggy, we were then told to delete ODFB and install the newer OneDrive app instead and things have been far more stable since then. Might be worth a try adding the OneDrive app into the mix and seeing if it helps?