diff --git a/src/components/EditorHeader/ControlPanel.jsx b/src/components/EditorHeader/ControlPanel.jsx
index 02d95fdd..ed17920f 100644
--- a/src/components/EditorHeader/ControlPanel.jsx
+++ b/src/components/EditorHeader/ControlPanel.jsx
@@ -767,6 +767,7 @@ export default function ControlPanel({
           setTables(diagram.tables);
           setRelationships(diagram.references);
           setAreas(diagram.areas);
+          setGistId(diagram.gistId ?? "");
           setNotes(diagram.notes);
           setTasks(diagram.todos ?? []);
           setTransform({
diff --git a/src/components/EditorHeader/Modal/Modal.jsx b/src/components/EditorHeader/Modal/Modal.jsx
index ca4ff819..c31e52ad 100644
--- a/src/components/EditorHeader/Modal/Modal.jsx
+++ b/src/components/EditorHeader/Modal/Modal.jsx
@@ -8,7 +8,7 @@ import {
 import { saveAs } from "file-saver";
 import { Parser } from "node-sql-parser";
 import { Parser as OracleParser } from "oracle-sql-parser";
-import { useState } from "react";
+import { useContext, useState } from "react";
 import { useTranslation } from "react-i18next";
 import { DB, MODAL, STATUS, State } from "../../../data/constants";
 import { databases } from "../../../data/databases";
@@ -40,6 +40,7 @@ import Open from "./Open";
 import Rename from "./Rename";
 import SetTableWidth from "./SetTableWidth";
 import Share from "./Share";
+import { IdContext } from "../../Workspace";
 
 const extensionToLanguage = {
   md: "markdown",
@@ -60,6 +61,7 @@ export default function Modal({
   importFrom,
 }) {
   const { t, i18n } = useTranslation();
+  const { setGistId } = useContext(IdContext);
   const { setTables, setRelationships, database, setDatabase } = useDiagram();
   const { setNotes } = useNotes();
   const { setAreas } = useAreas();
@@ -119,6 +121,7 @@ export default function Modal({
           setAreas(diagram.areas);
           setNotes(diagram.notes);
           setTasks(diagram.todos ?? []);
+          setGistId(diagram.gistId ?? "");
           setTransform({
             pan: diagram.pan,
             zoom: diagram.zoom,