diff --git a/sheditor.py b/sheditor.py
index 22fa8df..9cd3ab5 100644
--- a/sheditor.py
+++ b/sheditor.py
@@ -584,7 +584,7 @@ class GameData:
 
         # Step 3 - Update research
         console.print('Updating research')
-        top_research_tag = self.soup.find('research', treeId=True)
+        top_research_tag = self.soup.find('research', treeId=True).find('states')
         top_research_tag.clear()
         for research_item in sorted(self.research, key=lambda x:x.id):
             base_tag = self.soup.new_tag('l', attrs={'techId':research_item.id, 'paused':research_item.paused, 'activeStageIndex':research_item.active_stage})