diff --git a/src-ui/src/app/components/common/custom-fields-query-dropdown/custom-fields-query-dropdown.component.spec.ts b/src-ui/src/app/components/common/custom-fields-query-dropdown/custom-fields-query-dropdown.component.spec.ts
index ae54b85ba..7062be77a 100644
--- a/src-ui/src/app/components/common/custom-fields-query-dropdown/custom-fields-query-dropdown.component.spec.ts
+++ b/src-ui/src/app/components/common/custom-fields-query-dropdown/custom-fields-query-dropdown.component.spec.ts
@@ -266,7 +266,7 @@ describe('CustomFieldsQueryDropdownComponent', () => {
     await fixture.whenStable()
     expect(atom.value).toBeNull()
     const amount: HTMLInputElement = query('input[aria-label="Amount"]')
-    const unit: HTMLSelectElement = query('select[aria-label="Unit"]')
+    const unit: HTMLSelectElement = query('select[id$="_unit"]')
     amount.value = '30'
     amount.dispatchEvent(new Event('input'))
     expect(atom.serialize()).toEqual([4, 'gte', '30 days ago'])