Fix "flat is not a function" error

v9hstk · Dec 17, 2019 · 457f187 · 457f187
1 parent 7402d3b
commit 457f187
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/lib/rules/best-practises/max-states-count.js b/lib/rules/best-practises/max-states-count.js
@@ -1,3 +1,4 @@
+const _ = require('lodash')
 const BaseChecker = require('./../base-checker')
 const { severityDescription } = require('../../doc/utils')
 
@@ -59,10 +60,10 @@ class MaxStatesCountChecker extends BaseChecker {
   }
 
   ContractDefinition(node) {
-    const countOfVars = node.subNodes
+    const countOfVars = _(node.subNodes)
       .filter(({ type }) => type === 'StateVariableDeclaration')
-      .map(subNode => subNode.variables.filter(variable => !variable.isDeclaredConst))
-      .flat().length
+      .flatMap(subNode => subNode.variables.filter(variable => !variable.isDeclaredConst))
+      .value().length
 
     if (countOfVars > this.maxStatesCount) {
       this._error(node, countOfVars)