Ability to fly persisting when it shouldn't

So I made an effect that lets you fly around just like in creative mode. It works fine while it's on, but when the effect expires, the ability to fly persists until I quit and rejoin the world, while it should go away as soon as the effect itself wears off. I have this for the on effect active tick procedure - https://imgur.com/a/KYzP8SG

Any help with fixing this would be appreciated.