Correct tree level counting