diff --git a/lib/worker/setter.dart b/lib/worker/setter.dart
index c2a3d44..50e9b49 100644
--- a/lib/worker/setter.dart
+++ b/lib/worker/setter.dart
@@ -42,7 +42,7 @@ void setModel(BuildContext context, Function setState) {
         models.add(list.models![i].model!.split(":")[0]);
         modelsReal.add(list.models![i].model!);
         modal.add((list.models![i].details!.families ?? []).contains("clip") ||
-            (details.capabilities ?? []).contains("vision"));
+            (details.capabilities ?? []).contains(llama.Capability.vision));
       }
 
       addIndex = models.length;