diff --git a/capabilities/classification.mdx b/capabilities/classification.mdx
index b251d89..77635a2 100644
--- a/capabilities/classification.mdx
+++ b/capabilities/classification.mdx
@@ -14,6 +14,8 @@ from tabpfn import TabPFNClassifier
 from sklearn.datasets import load_breast_cancer
 from sklearn.model_selection import train_test_split
 from sklearn.metrics import accuracy_score, log_loss
+# To use TabPFNv2:
+# from tabpfn.constants import ModelVersion
 
 X, y = load_breast_cancer(return_X_y=True)
 X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_state=42)
@@ -44,4 +46,4 @@ print("Log-Loss:", log_loss(y_test, probs))
   <Card title="Metric Tuning" icon="puzzle" href="/capabilities/metric-tuning">
     Tune scores other than maximum probability prediction such as F1 or balanced accuracy.
   </Card>
-</CardGroup>
\ No newline at end of file
+</CardGroup>