diff --git a/ui-ngx/src/app/modules/home/components/calculated-fields/calculated-fields-table-config.ts b/ui-ngx/src/app/modules/home/components/calculated-fields/calculated-fields-table-config.ts
index 37dc5d69e7..cc4a100b2b 100644
--- a/ui-ngx/src/app/modules/home/components/calculated-fields/calculated-fields-table-config.ts
+++ b/ui-ngx/src/app/modules/home/components/calculated-fields/calculated-fields-table-config.ts
@@ -263,18 +263,17 @@ export class CalculatedFieldsTableConfig extends EntityTableConfig<CalculatedFie
   }
 
   private updateImportedCalculatedField(calculatedField: CalculatedField): CalculatedField {
+    const updatedArguments = Object.keys(calculatedField.configuration.arguments).reduce((acc, key) => {
+      const arg = calculatedField.configuration.arguments[key];
+      acc[key] = arg.refEntityId?.entityType === ArgumentEntityType.Tenant
+        ? { ...arg, refEntityId: { id: this.tenantId, entityType: ArgumentEntityType.Tenant } }
+        : arg;
+      return acc;
+    }, {});
+
     return {
       ...calculatedField,
-      configuration: {
-        ...calculatedField.configuration,
-        arguments: Object.keys(calculatedField.configuration.arguments).reduce((acc, key) => {
-          const arg = calculatedField.configuration.arguments[key];
-          acc[key] = arg.refEntityId?.entityType === ArgumentEntityType.Tenant
-            ? { ...arg, refEntityId: { id: this.tenantId, entityType: ArgumentEntityType.Tenant } }
-            : arg;
-          return acc;
-        }, {})
-      }
+      configuration: { ...calculatedField.configuration, arguments: updatedArguments }
     }
   }