diff --git a/pwa/src/components/PatientInfoForm.test.tsx b/pwa/src/components/PatientInfoForm.test.tsx
index 9f0f495..23d175b 100644
--- a/pwa/src/components/PatientInfoForm.test.tsx
+++ b/pwa/src/components/PatientInfoForm.test.tsx
@@ -96,7 +96,9 @@ test('sets albendazole input', () => {
             <PatientInfoForm />
         </StoreProvider>,
     )
-    const femaleButton = dom.getByTestId('female-button')
+
+    const albendazoleYesInput = dom.queryByTestId('albendazole-Yes')
+    const checkBox = albendazoleYesInput.getElementsByTagName('span')[0]
 
     expect(checkBox).not.toHaveClass('am-radio-checked')
     fireEvent.click(albendazoleYesInput)