From 44d612488be69486ab82b9ee1e8d21512d1a36fa Mon Sep 17 00:00:00 2001
From: Ibitoye Rotimi Best <best.rotimi@ikizmet.com>
Date: Tue, 6 Jul 2021 08:49:51 +0300
Subject: [PATCH] After save undo preview

---
 .../Course/components/Lecture/Homework/index.svelte        | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/components/Course/components/Lecture/Homework/index.svelte b/src/components/Course/components/Lecture/Homework/index.svelte
index 0faf169b..0f09afb0 100644
--- a/src/components/Course/components/Lecture/Homework/index.svelte
+++ b/src/components/Course/components/Lecture/Homework/index.svelte
@@ -9,6 +9,11 @@
 
   let mode = MODES.view;
   let preview = false;
+
+  function handleMode() {
+    mode = mode === MODES.edit ? MODES.view : MODES.edit;
+    preview = false;
+  }
 </script>
 
 <PageBody>
@@ -18,7 +23,7 @@
         className="mr-2"
         variant={VARIANTS.OUTLINED}
         label={mode === MODES.edit ? "Save" : "Edit"}
-        onClick={() => (mode = mode === MODES.edit ? MODES.view : MODES.edit)}
+        onClick={handleMode}
       />
       {#if mode === MODES.edit || preview}
         <PrimaryButton