Fix TSTree so it properly keeps track of its size