1 files changed, 2 insertions, 1 deletions
diff --git a/catalog-ui/src/app/ng2/pages/properties-assignment/tosca-function/tosca-custom-function/tosca-custom-function.component.ts b/catalog-ui/src/app/ng2/pages/properties-assignment/tosca-function/tosca-custom-function/tosca-custom-function.component.ts
index ad72adad22..7db332236b 100644
--- a/catalog-ui/src/app/ng2/pages/properties-assignment/tosca-function/tosca-custom-function/tosca-custom-function.component.ts
+++ b/catalog-ui/src/app/ng2/pages/properties-assignment/tosca-function/tosca-custom-function/tosca-custom-function.component.ts
@@ -49,6 +49,7 @@ export class ToscaCustomFunctionComponent implements OnInit {
     @Input() propertyType: string;
     @Input() propertySchemaType: string = undefined;
     @Input() isDefaultCustomFunction: boolean;
+    @Input() overridingType: PROPERTY_TYPES;
     @Output() onValidFunction: EventEmitter<ToscaCustomFunction> = new EventEmitter<ToscaCustomFunction>();
     @Output() onValidityChange: EventEmitter<ToscaCustomFunctionValidationEvent> = new EventEmitter<ToscaCustomFunctionValidationEvent>();
 
@@ -194,7 +195,7 @@ export class ToscaCustomFunctionComponent implements OnInit {
     createProperty(value?: any): PropertyBEModel {
         const property = new PropertyBEModel();
         if (this.type === this.GET_INPUT) {
-            property.type = this.propertyType;
+            property.type = this.overridingType ? this.overridingType.toString() : this.propertyType;
             if (this.propertySchemaType) {
                 property.schemaType = this.propertySchemaType;
             }