diff --git a/src/Umbraco.Web.UI.Client/src/backoffice/core/property-editors/uis/date-picker/property-editor-ui-date-picker.test.ts b/src/Umbraco.Web.UI.Client/src/backoffice/core/property-editors/uis/date-picker/property-editor-ui-date-picker.test.ts
index 11a4002b8f..901e4ac1d2 100644
--- a/src/Umbraco.Web.UI.Client/src/backoffice/core/property-editors/uis/date-picker/property-editor-ui-date-picker.test.ts
+++ b/src/Umbraco.Web.UI.Client/src/backoffice/core/property-editors/uis/date-picker/property-editor-ui-date-picker.test.ts
@@ -1,15 +1,15 @@
 import { expect, fixture, html } from '@open-wc/testing';
-import { UUIInputElement } from '@umbraco-ui/uui';
+import { UmbDateInputElement } from '../../../components/date-input/date-input.element';
 import { UmbPropertyEditorUIDatePickerElement } from './property-editor-ui-date-picker.element';
 import { defaultA11yConfig } from '@umbraco-cms/internal/test-utils';
 
 describe('UmbPropertyEditorUIDatePickerElement', () => {
 	let element: UmbPropertyEditorUIDatePickerElement;
-	let inputElement: UUIInputElement;
+	let inputElement: UmbDateInputElement;
 
 	beforeEach(async () => {
 		element = await fixture(html` <umb-property-editor-ui-date-picker></umb-property-editor-ui-date-picker> `);
-		inputElement = element.shadowRoot?.querySelector('uui-input') as UUIInputElement;
+		inputElement = element.shadowRoot?.querySelector('umb-date-input') as UmbDateInputElement;
 	});
 
 	it('is defined with its own instance', () => {