diff --git a/tests/spec/text-input_spec.js b/tests/spec/text-input_spec.js
index af08b9f350c9..0d590151a274 100644
--- a/tests/spec/text-input_spec.js
+++ b/tests/spec/text-input_spec.js
@@ -51,7 +51,7 @@ describe('Test text input', () => {
     beforeAll(() => {
       document.body.appendChild(container);
       new TextInput(document.querySelector('[data-text-input]'));
-      textInput = document.querySelector('.bx--form-item');
+      textInput = document.querySelector('[data-text-input]');
       passwordVisibilityButton = document.querySelector('.bx--text-input--password__visibility');
     });