diff --git a/3d_segmentation/unetr_btcv_segmentation_3d.ipynb b/3d_segmentation/unetr_btcv_segmentation_3d.ipynb
index 6654f44e2c..264f3676ff 100644
--- a/3d_segmentation/unetr_btcv_segmentation_3d.ipynb
+++ b/3d_segmentation/unetr_btcv_segmentation_3d.ipynb
@@ -586,7 +586,7 @@
     "    hidden_size=768,\n",
     "    mlp_dim=3072,\n",
     "    num_heads=12,\n",
-    "    pos_embed=\"perceptron\",\n",
+    "    proj_type=\"perceptron\",\n",
     "    norm_name=\"instance\",\n",
     "    res_block=True,\n",
     "    dropout_rate=0.0,\n",
diff --git a/3d_segmentation/unetr_btcv_segmentation_3d_lightning.ipynb b/3d_segmentation/unetr_btcv_segmentation_3d_lightning.ipynb
index bc5b39a08f..3d383bd78d 100644
--- a/3d_segmentation/unetr_btcv_segmentation_3d_lightning.ipynb
+++ b/3d_segmentation/unetr_btcv_segmentation_3d_lightning.ipynb
@@ -423,7 +423,7 @@
     "            hidden_size=768,\n",
     "            mlp_dim=3072,\n",
     "            num_heads=12,\n",
-    "            pos_embed=\"perceptron\",\n",
+    "            proj_type=\"perceptron\",\n",
     "            norm_name=\"instance\",\n",
     "            res_block=True,\n",
     "            conv_block=True,\n",
diff --git a/deepedit/ignite/train.py b/deepedit/ignite/train.py
index 871b574cf2..ae452d6ce4 100644
--- a/deepedit/ignite/train.py
+++ b/deepedit/ignite/train.py
@@ -76,7 +76,7 @@ def get_network(network, labels, spatial_size):
             hidden_size=1536,
             mlp_dim=3072,
             num_heads=48,
-            pos_embed="conv",
+            proj_type="conv",
             norm_name="instance",
             res_block=True,
         )
diff --git a/self_supervised_pretraining/vit_unetr_ssl/multi_gpu/mgpu_ssl_train.py b/self_supervised_pretraining/vit_unetr_ssl/multi_gpu/mgpu_ssl_train.py
index 52499756a5..3364f04925 100644
--- a/self_supervised_pretraining/vit_unetr_ssl/multi_gpu/mgpu_ssl_train.py
+++ b/self_supervised_pretraining/vit_unetr_ssl/multi_gpu/mgpu_ssl_train.py
@@ -175,7 +175,7 @@ def main(args):
         in_channels=1,
         img_size=(96, 96, 96),
         patch_size=(16, 16, 16),
-        pos_embed="conv",
+        proj_type="conv",
         hidden_size=768,
         mlp_dim=3072,
     )
diff --git a/self_supervised_pretraining/vit_unetr_ssl/ssl_finetune.ipynb b/self_supervised_pretraining/vit_unetr_ssl/ssl_finetune.ipynb
index 68e14ebaa7..d4e9742cf2 100644
--- a/self_supervised_pretraining/vit_unetr_ssl/ssl_finetune.ipynb
+++ b/self_supervised_pretraining/vit_unetr_ssl/ssl_finetune.ipynb
@@ -317,7 +317,7 @@
     "    hidden_size=768,\n",
     "    mlp_dim=3072,\n",
     "    num_heads=12,\n",
-    "    pos_embed=\"conv\",\n",
+    "    proj_type=\"conv\",\n",
     "    norm_name=\"instance\",\n",
     "    res_block=True,\n",
     "    dropout_rate=0.0,\n",
diff --git a/self_supervised_pretraining/vit_unetr_ssl/ssl_train.ipynb b/self_supervised_pretraining/vit_unetr_ssl/ssl_train.ipynb
index bada2138a0..c730415b4c 100644
--- a/self_supervised_pretraining/vit_unetr_ssl/ssl_train.ipynb
+++ b/self_supervised_pretraining/vit_unetr_ssl/ssl_train.ipynb
@@ -253,7 +253,7 @@
     "    in_channels=1,\n",
     "    img_size=(96, 96, 96),\n",
     "    patch_size=(16, 16, 16),\n",
-    "    pos_embed=\"conv\",\n",
+    "    proj_type=\"conv\",\n",
     "    hidden_size=768,\n",
     "    mlp_dim=3072,\n",
     ")\n",