diff --git a/src/components/Workspace.jsx b/src/components/Workspace.jsx
index ebba0f59..fd24554c 100644
--- a/src/components/Workspace.jsx
+++ b/src/components/Workspace.jsx
@@ -290,6 +290,7 @@ export default function WorkSpace() {
         const res = await get(shareId);
         const diagramSrc = res.data.files["share.json"].content;
         const d = JSON.parse(diagramSrc);
+        setGistId(shareId);
         setUndoStack([]);
         setRedoStack([]);
         setLoadedFromGistId(shareId);