Use the correct tree view class