1 files changed, 3 insertions, 0 deletions
diff --git a/src/character_system.cpp b/src/character_system.cpp
index d18c592..a24bdcc 100644
--- a/src/character_system.cpp
+++ b/src/character_system.cpp

@@ -46,10 +46,12 @@ void CharacterSystem::addSpriteToParty(int leaderId, int followerId) {
 void CharacterSystem::transplantParty(int leaderId, vec2i pos, Direction dir) {
  Sprite& leader = game_.getSprite(leaderId);
  CharacterState oldState = leader.characterState;
+  CharacterMedium newMedium = game_.getSystem<TransformSystem>().getMediumAtPosition(leaderId, pos);
  std::vector<int> followers = leader.followers;
  leader.followers.clear();
  leader.trailsAreHalved = false;
+  leader.characterMedium = newMedium;
  game_.getSystem<TransformSystem>().moveSprite(leaderId, pos);
  game_.getSystem<AnimationSystem>().setSpriteDirection(leaderId, dir);
@@ -60,6 +62,7 @@ void CharacterSystem::transplantParty(int leaderId, vec2i pos, Direction dir) {
    game_.getSystem<TransformSystem>().moveSprite(followerId, pos);
    game_.getSystem<AnimationSystem>().setSpriteDirection(followerId, dir);
+    follower.characterMedium = newMedium;
    addSpriteToParty(leaderId, followerId);
  }