Detect block type and move a mob

To fix a mob I have sitting on top of the water and losing it's AI I figured testing for air blocks above it and teleporting it down a few blocks ( back into the water ) would work? Just not sure how to go about doing that