diff --git a/part4/4.JetTaggingPointCloud.ipynb b/part4/4.JetTaggingPointCloud.ipynb
index b1fea0d..7fd47d0 100644
--- a/part4/4.JetTaggingPointCloud.ipynb
+++ b/part4/4.JetTaggingPointCloud.ipynb
@@ -16,7 +16,7 @@
         "id": "BsonEqBekjyy"
       },
       "source": [
-        "# Jet Tagging with Set Transformer\n",
+        "# Jet Tagging with Permutation Invariance\n",
         "\n",
         "In this notebook we will see an implementation of the Transformer architecture for sets applied to the jet tagging task. For *sets* it is meant here a point cloud, i.e. a set of nodes without edges. We will instead use Multi-Head Attention to learn which nodes (or particles) have strong pair-wise interaction.\n",
         "\n",