Fix working with Tree Style Tab etc.