diff --git a/frontend/src/app/main/data/workspace/colors.cljs b/frontend/src/app/main/data/workspace/colors.cljs
index 287b6c3447..5e9028424a 100644
--- a/frontend/src/app/main/data/workspace/colors.cljs
+++ b/frontend/src/app/main/data/workspace/colors.cljs
@@ -602,9 +602,11 @@
                     (update-in state [:stops stop] (fn [data] (->> changes
                                                                    (merge data)
                                                                    (materialize-color-components))))
+
                     (-> state
                         (dissoc :gradient :stops :editing-stop)
-                        (assoc :type :color)))))))
+                        (cond-> (not= :image (:type state))
+                          (assoc :type :color))))))))
     ptk/WatchEvent
     (watch [_ state _]
       (when add-recent?