Whenever I put several kayentatherium into enclosure, after they wake up and go for a swim they won’t get out of water to breathe but they keep diving until they run out off oxygen and take damage until they died, so I think you should make them can breathe underwater in the code like calsoyasuchus to fix this problem