diff --git a/lib/modules/graph/index.js b/lib/modules/graph/index.js
index 21d6ce32d..5ccfe2952 100644
--- a/lib/modules/graph/index.js
+++ b/lib/modules/graph/index.js
@@ -3,7 +3,7 @@ const Viz = require('viz.js');
 const fs = require('fs');
 
 class GraphGenerator {
-  constructor(embark, options) {
+  constructor(embark, _options) {
     const self = this;
     this.events = embark.events;
     this.contracts = [];
@@ -145,7 +145,7 @@ class GraphGenerator {
   }
 
   getContract(contractName) {
-    return this.contracts.find((contract) => { return contract.className === contractName });
+    return this.contracts.find((contract) => { return contract.className === contractName; });
   }
 }