Sugar cane grow in water but cant be waterlogged 

I think there's two thing you can do to repair this,

or make that sugar cane can waterlog as normal sugar can grow in water

or make that sugar cane break when put next to water