JQuery :: TreeView - Lose Persistence Of Location

I load the content from json, but I lose persistence of location. I get the tree folded and I don't know what branch I'm on to. I managed to fix this by pasting the location code in .async.js,
var sm = current.find("a").filter(function() { return
this.href.toLowerCase() == location.href.toLowerCase(); });
if ( sm.length ) {
sm.addClass("selected").parents
("ul, li").add( sm.next() ).show
();
}
But now the branch is marked as selected, however the tree is not open (all folded).