I can see why it eventually became fully aquatic, as it already looks semi-aquatic. Based on its shape, I can imagine it standing in shallow water like a hippo with its head floating at the surface of the water waiting to strike like a crocodile.