All axolotls I have spawned eventually find their way onto the extinguished campfires and then either spin around or don't move at all until they suffocate, they never seem to make an attempt to return to the water. The level of the campfire is 0.5 blocks above the water level and there is water below it.