rewrite run_unit for drone support

[#78964864]