diff --git a/ayanova/src/components/work-order-item-units.vue b/ayanova/src/components/work-order-item-units.vue
index 4f448910..b645dbb7 100644
--- a/ayanova/src/components/work-order-item-units.vue
+++ b/ayanova/src/components/work-order-item-units.vue
@@ -157,9 +157,13 @@
             </v-card-text>
             <v-card-actions>
               <v-spacer></v-spacer>
-              <v-btn color="primary" text @click="getWarrantyInfo">{{
-                $ay.t("Search")
-              }}</v-btn>
+              <v-btn
+                color="primary"
+                text
+                @click="getWarrantyInfo"
+                data-cy="woItemUnitGetWarrantyInfo"
+                >{{ $ay.t("Search") }}</v-btn
+              >
             </v-card-actions>
           </v-card>
         </v-col>
diff --git a/ayanova/src/components/work-order-items.vue b/ayanova/src/components/work-order-items.vue
index 38da954a..979bf44f 100644
--- a/ayanova/src/components/work-order-items.vue
+++ b/ayanova/src/components/work-order-items.vue
@@ -62,6 +62,7 @@
               <v-list-item
                 v-if="canAddUnit"
                 @click="newSubItem($ay.ayt().WorkOrderItemUnit)"
+                data-cy="woItemAddUnit"
               >
                 <v-list-item-icon>
                   <v-icon>$ayiFan</v-icon>