I was thinking along those lines, especially concerning sea snakes /kraits but I don't really know too much about them. I mean do rattle snakes swim? As they all have keeled scales as far as I'm aware.

On a side note I prefer keeled scales, I find they feel better when I'm handling than with smooth scaled snakes.