diff --git a/src/SaveTool/SaveTool_MainManager.cpp b/src/SaveTool/SaveTool_MainManager.cpp
index 78eda67..97018fa 100644
--- a/src/SaveTool/SaveTool_MainManager.cpp
+++ b/src/SaveTool/SaveTool_MainManager.cpp
@@ -402,8 +402,10 @@ void SaveTool::drawResearchTree() {
         _tree->readEngineUnlocks(_currentProfile->engineInventory(), _currentProfile->gearInventory());
 
         _tree->generateOSTree();
+        _tree->readOSUnlocks(_currentProfile->osInventory(), _currentProfile->moduleInventory());
 
         _tree->generateArchTree();
+        _tree->readArchUnlocks(_currentProfile->archInventory(), _currentProfile->techInventory());
     }
 
     if(ImGui::BeginTabBar("##TreeTabBar")) {