Fixed #106

primefaces · Sep 21, 2017 · 603df83 · 603df83
1 parent 77e5b11
commit 603df83
Showing 1 changed file with 14 additions and 14 deletions.
diff --git a/src/components/datatable/DataTable.js b/src/components/datatable/DataTable.js
@@ -733,31 +733,31 @@ export class DataTable extends Component {
     }
 
     getFrozenColumns() {
+        let columns = React.Children.toArray(this.props.children);
         let frozenColumns = null;
-        if(this.props.children && this.props.children.length) {
-            for(let col of this.props.children) {
-                if(col.props.frozen) {
-                    frozenColumns = frozenColumns||[];
-                    frozenColumns.push(col);
-                }
+
+        for(let col of columns) {
+            if(col.props.frozen) {
+                frozenColumns = frozenColumns||[];
+                frozenColumns.push(col);
             }
         }
 
         return frozenColumns;
     }
 
     getScrollableColumns() {
-        let columns = null;
-        if(this.props.children && this.props.children.length) {
-            for(let col of this.props.children) {
-                if(!col.props.frozen) {
-                    columns = columns||[];
-                    columns.push(col);
-                }
+        let columns = React.Children.toArray(this.props.children);
+        let scrollableColumns = null;
+
+        for(let col of columns) {
+            if(!col.props.frozen) {
+                scrollableColumns = scrollableColumns||[];
+                scrollableColumns.push(col);
             }
         }
 
-        return columns;
+        return scrollableColumns;
     }
 
     createTableHeader(columns) {