small fix for tree store