From 9538f65d53167f3adee67ef4d82ce1019a4d9f1f Mon Sep 17 00:00:00 2001
From: Vladyslav <vprykhodko@thingsboard.io>
Date: Tue, 10 Sep 2019 15:43:11 +0300
Subject: [PATCH] Fix datakey function (#1992)

---
 ui/src/app/components/datasource-func.directive.js | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/ui/src/app/components/datasource-func.directive.js b/ui/src/app/components/datasource-func.directive.js
index 91d43f5bb6..1e651b947f 100644
--- a/ui/src/app/components/datasource-func.directive.js
+++ b/ui/src/app/components/datasource-func.directive.js
@@ -183,9 +183,9 @@ function DatasourceFunc($compile, $templateCache, $mdDialog, $window, $document,
                     w.triggerHandler('resize');
                 }
             }).then(function (newDataKey) {
-                if ((newDataKey.type === types.dataKeyType.timeseries) || (newDataKey.type === types.dataKeyType.attribute)) {
-                    let index = scope.dataKeys.indexOf(dataKey);
-                    scope.dataKeys[index] = newDataKey;
+                if (newDataKey.type === types.dataKeyType.function) {
+                    let index = scope.funcDataKeys.indexOf(dataKey);
+                    scope.funcDataKeys[index] = newDataKey;
                 } else if (newDataKey.type === types.dataKeyType.alarm) {
                     let index = scope.alarmDataKeys.indexOf(dataKey);
                     scope.alarmDataKeys[index] = newDataKey;