diff --git a/src/Umbraco.Web.UI.Client/src/core/modal/layouts/content-picker/modal-layout-content-picker.element.ts b/src/Umbraco.Web.UI.Client/src/core/modal/layouts/content-picker/modal-layout-content-picker.element.ts
index ae0c7f46ed..2ae50a2c6e 100644
--- a/src/Umbraco.Web.UI.Client/src/core/modal/layouts/content-picker/modal-layout-content-picker.element.ts
+++ b/src/Umbraco.Web.UI.Client/src/core/modal/layouts/content-picker/modal-layout-content-picker.element.ts
@@ -55,18 +55,20 @@ export class UmbModalLayoutContentPickerElement extends UmbModalLayoutElement<Um
 	@state()
 	_alias = '';
 
+	@state()
+	_multiple = true;
+
 	connectedCallback() {
 		super.connectedCallback();
 		this._selection = this.data?.selection ?? [];
 		this._alias = this.data?.alias ?? 'Umb.Tree.Documents';
+		this._multiple = this.data?.multiple ?? true;
 	}
 
 	private _handleSelectionChange(e: CustomEvent) {
 		e.stopPropagation();
-		console.log('Selection change', e);
-
 		const element = e.target as UmbTreeElement;
-		this._selection = element.selection;
+		this._selection = this._multiple ? element.selection : [element.selection[element.selection.length - 1]];
 	}
 
 	private _submit() {