So when you're underwater, the on_ground condition does the opposite of itself, so if you were trying to detect you being on the ocean floor, it would think you're not, and vice versa if you put a not condition around it

a temporary fix could be using a command condition to check if there's a block under you

Yeah, that's probably what I'll do