diff --git a/core/src/mindustry/input/InputHandler.java b/core/src/mindustry/input/InputHandler.java
index 728c17b3b3..f20577ab06 100644
--- a/core/src/mindustry/input/InputHandler.java
+++ b/core/src/mindustry/input/InputHandler.java
@@ -1211,7 +1211,7 @@ public abstract class InputHandler implements InputProcessor, GestureListener{
                     var cmd =  ai.currentCommand();
                     lastPos = ai.attackTarget != null ? ai.attackTarget : ai.targetPos;
 
-                    if(unit.isFlying() != flying) continue;
+                    if((unit.isFlying() || unit.type.allowLegStep) != flying) continue;
 
                     //draw target line
                     if(ai.targetPos != null && cmd.drawTarget){