[keep-tree-state] Fixed broken tests