diff --git a/src/MissionManager/MissionCommandTree.cc b/src/MissionManager/MissionCommandTree.cc
index fa74a70491eacdbdbc80a91fa5cefefad3137655..e6828007b3267cc136a5d20df6a32a398f5adc33 100644
--- a/src/MissionManager/MissionCommandTree.cc
+++ b/src/MissionManager/MissionCommandTree.cc
@@ -102,7 +102,7 @@ void MissionCommandTree::_collapseHierarchy(Vehicle*
 
     foreach (MAV_CMD command, cmdList->commandIds()) {
         // Only add supported command to tree (MAV_CMD_NAV_LAST is used for planned home position)
-        if (!vehicle->firmwarePlugin()->supportedMissionCommands().contains(command) && command != MAV_CMD_NAV_LAST) {
+        if (!qgcApp()->runningUnitTests() && !vehicle->firmwarePlugin()->supportedMissionCommands().contains(command) && command != MAV_CMD_NAV_LAST) {
             continue;
         }