diff --git a/setup.py b/setup.py
index 00a4a08..93f6e82 100644
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@
 
 setup(
     name='TraverseCraft',
-    version='1.0.0',
+    version='1.1.0',
     author='Srajan Chourasia, Varun Patrikar',
     author_email='srajanstark.ash@gmail.com, patrikarvarun@gmail.com',
     maintainer='Srajan Chourasia, Varun Patrikar',
diff --git a/src/traverseCraft/world.py b/src/traverseCraft/world.py
index 654258d..975a908 100644
--- a/src/traverseCraft/world.py
+++ b/src/traverseCraft/world.py
@@ -613,7 +613,7 @@ def _check_tree_format(self,treeWorldInfo):
         # Ensure all nodes in 'adj' and 'goals' are in 'position'
         position_keys = set(position.keys())
         adj_keys = set(adj.keys())
-        root_key = set(treeWorldInfo['root'])
+        root_key = set([treeWorldInfo['root']])
         goals_set = set(goals)
         
         if not root_key.issubset(position_keys):