From c74d174b045e76b110cb1bd0176bee6c89c68945 Mon Sep 17 00:00:00 2001
From: Yan Chen <yan.chen@dfinity.org>
Date: Thu, 7 Sep 2023 17:57:30 -0700
Subject: [PATCH] fix

---
 src/components/CanisterModal.tsx |  4 ++++
 src/components/ImportGithub.tsx  | 10 +++++++++-
 2 files changed, 13 insertions(+), 1 deletion(-)

diff --git a/src/components/CanisterModal.tsx b/src/components/CanisterModal.tsx
index 62a12f3f..c5d0e85b 100644
--- a/src/components/CanisterModal.tsx
+++ b/src/components/CanisterModal.tsx
@@ -108,6 +108,10 @@ export function CanisterModal({ isOpen, close, deploySetter }) {
     await deploySetter.setInitTypes(init);
     await deploySetter.setCandidCode(candid);
     await deploySetter.setShowDeployModal(true);
+    await dispatch({
+      type: "setOrigin",
+      payload: { origin: "playground:wasm", tags: [] },
+    });
   }
   async function addCanister() {
     if (error || !canisterName || !canisterId || !candid) {
diff --git a/src/components/ImportGithub.tsx b/src/components/ImportGithub.tsx
index 07eb119f..c0f5e4fe 100644
--- a/src/components/ImportGithub.tsx
+++ b/src/components/ImportGithub.tsx
@@ -2,7 +2,10 @@ import styled from "styled-components";
 import { useState, useContext } from "react";
 import { Button } from "./shared/Button";
 import { PackageInfo } from "../workers/file";
-import { WorkerContext } from "../contexts/WorkplaceState";
+import {
+  WorkerContext,
+  WorkplaceDispatchContext,
+} from "../contexts/WorkplaceState";
 import { Field } from "./shared/Field";
 
 const Container = styled.div`
@@ -33,12 +36,17 @@ export function ImportGitHub({ importCode, close, isPackageModal = false }) {
   const [error, setError] = useState("");
   const [name, setName] = useState("");
   const worker = useContext(WorkerContext);
+  const dispatch = useContext(WorkplaceDispatchContext);
   async function fetchCode() {
     const files = await worker.fetchGithub({ repo, branch, dir });
     if (files) {
       setError("");
       importCode(files);
       close();
+      await dispatch({
+        type: "setOrigin",
+        payload: { origin: "playground:git", tags: [`git:${repo}`] },
+      });
     } else {
       setError(`Cannot find repo or the directory contains no ".mo" files.`);
     }