How do i keep my hair straight while swimming? ?

I like to straighten my hair alot bcuz my hair is naturally curly and frizzy but I also like swimming alot but my problem is that whenever I straighten my hair then get it wet it goes back to curly and frizzy, so is there a way to keep it straight while im in the water?