diff --git a/src/stores/generator.ts b/src/stores/generator.ts
index 2585cfb2..8ba02639 100644
--- a/src/stores/generator.ts
+++ b/src/stores/generator.ts
@@ -193,13 +193,14 @@ export const useGeneratorStore = defineStore("generator", () => {
     const filteredAvailableModels = computed(() => {
         if (availableModels.value.length === 0) return [];
         let filtered = availableModels.value.filter(el => {
+            const isInpaintingModel = el.value.toLowerCase().includes("inpainting")
             if (generatorType.value === "Inpainting") {
-                return el.value.includes("inpainting") && el.value !== "Stable Diffusion 2 Depth";
+                return isInpaintingModel && el.value !== "Stable Diffusion 2 Depth";
             }
             if (generatorType.value === "Img2Img") {
-                return el.value !== "stable_diffusion_2.0" && !el.value.includes("inpainting");
+                return el.value !== "stable_diffusion_2.0" && !isInpaintingModel;
             }
-            return !el.value.includes("inpainting") && el.value !== "pix2pix" && el.value !== "Stable Diffusion 2 Depth";
+            return !isInpaintingModel && el.value !== "pix2pix" && el.value !== "Stable Diffusion 2 Depth";
         });
         if (!filtered.find(el => el.value === selectedModel.value)) {
             selectedModel.value = filtered[0].value;