When in creative, if you attempt to fly while swimming (as in doing the swimming animation) you won't be able to fly. However if you stand in the water or out of it, you can easily fly.