diff --git a/src/views/CharacterDetailView.vue b/src/views/CharacterDetailView.vue
index a6cdfe5..0bf93f5 100644
--- a/src/views/CharacterDetailView.vue
+++ b/src/views/CharacterDetailView.vue
@@ -937,12 +937,13 @@ const handleLogout = () => {
                                                         @click.stop="reusePrompt(gen)"
                                                         v-tooltip.bottom="'Use this prompt'" />
                                                     <Button icon="pi pi-images" label="Asset" size="small" text
-                                                        class="!text-[10px] !py-0.5 !px-1.5 text-slate-400 hover:bg-white/5 flex-1"
+                                                        class="!text-[10px] !py-0.5 !px-2 text-slate-400 hover:bg-white/5 flex-1"
                                                         @click.stop="reuseAsset(gen)"
-                                                        v-tooltip.bottom="'Use original assets'" />
+                                                        v-tooltip.bottom="'Use original assets'"
+                                                        :disabled="gen.status !== 'done' || gen.assets_list.length == 0" />
                                                     <Button icon="pi pi-reply" label="Result" size="small" text
-                                                        class="!text-[10px] !py-0.5 !px-1.5 text-slate-400 hover:bg-white/5 flex-1"
-                                                        :disabled="gen.status !== 'done' || !gen.assets_list?.length"
+                                                        class="!text-[10px] !py-0.5 !px-2 text-slate-400 hover:bg-white/5 flex-1"
+                                                        :disabled="gen.status !== 'done' || gen.result_list.length == 0"
                                                         @click.stop="useResultAsReference(gen)"
                                                         v-tooltip.bottom="'Use result as reference'" />
                                                 </div>
diff --git a/src/views/ImageGenerationView.vue b/src/views/ImageGenerationView.vue
index 4c82e44..670cfe8 100644
--- a/src/views/ImageGenerationView.vue
+++ b/src/views/ImageGenerationView.vue
@@ -729,10 +729,10 @@ onMounted(() => {
                                         @click.stop="reusePrompt(gen)" v-tooltip.bottom="'Use this prompt'" />
                                     <Button icon="pi pi-images" label="Asset" size="small" text
                                         class="!text-[10px] !py-0.5 !px-2 text-slate-400 hover:bg-white/5 flex-1"
-                                        @click.stop="reuseAsset(gen)" v-tooltip.bottom="'Use original assets'" />
+                                        @click.stop="reuseAsset(gen)" v-tooltip.bottom="'Use original assets'" :disabled="gen.status !== 'done' || gen.assets_list.length==0" />
                                     <Button icon="pi pi-reply" label="Result" size="small" text
                                         class="!text-[10px] !py-0.5 !px-2 text-slate-400 hover:bg-white/5 flex-1"
-                                        :disabled="gen.status !== 'done' || !gen.assets_list?.length"
+                                        :disabled="gen.status !== 'done' || gen.result_list.length==0"
                                         @click.stop="useResultAsReference(gen)"
                                         v-tooltip.bottom="'Use result as reference'" />
                                 </div>