less instant disconnect, more broken multiplayer

core/src/mindustry/entities/comp/UnitComp.java

@@ -127,12 +127,12 @@ abstract class UnitComp implements Healthc, Physicsc, Hitboxc, Statusc, Teamc, I
     public void afterSync(){
         //set up type info after reading
         setStats(this.type);
+        controller.unit(this);
     }
 
     @Override
     public void afterRead(){
-        //set up type info after reading
-        setStats(this.type);
+        afterSync();
     }
 
     @Override

core/src/mindustry/entities/units/AIController.java

@@ -49,6 +49,8 @@ public class AIController implements UnitController{
 
     @Override
     public void unit(Unitc unit){
+        if(this.unit == unit) return;
+
         this.unit = unit;
         init();
     }