diff --git a/pretrain/PyTorch/notebooks/BERT_Pretrain.ipynb b/pretrain/PyTorch/notebooks/BERT_Pretrain.ipynb
index 5b6d992..cf1dc93 100644
--- a/pretrain/PyTorch/notebooks/BERT_Pretrain.ipynb
+++ b/pretrain/PyTorch/notebooks/BERT_Pretrain.ipynb
@@ -78,7 +78,7 @@
    "outputs": [],
    "source": [
     "# Retrieve the workspace\n",
-    "ws = Workspace.setup()\n",
+    "ws = Workspace.create()\n",
     "\n",
     "# Print the workspace attributes\n",
     "print('Workspace name: ' + ws.name, \n",
@@ -427,4 +427,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 2
-}
\ No newline at end of file
+}