1.20.1 version of Always Fly doesn't quite work as well as 1.21.1's

[HyperMan19680]

I'm using this mod on 1.20.1 and 1.21.1 and I noticed the Always Fly is being disabled on 1.20.1 whenever I shift-land. Thus giving me issues with another mod that makes me phase through blocks like Spectator Mode with the MekaSuit. I could potentially fall through the bedrock floor this way if I'm not careful. So I'd like the Always Fly option to, well, always fly. Even if I've landed.

[HyperMan19680]

Found the source of the issue, strangely if in Creative Mode, it gets disabled when I land. But it works fine in Survival, from what testing I have given it.