diff --git a/dagre/dagre-tests.ts b/dagre/dagre-tests.ts
index ab790e502..621727944 100644
--- a/dagre/dagre-tests.ts
+++ b/dagre/dagre-tests.ts
@@ -1,13 +1,10 @@
 /// <reference path="dagre.d.ts"/>
 module DagreTests {
   var gDagre = new dagre.graphlib.Graph();
-  gDagre.setGraph({});
-  gDagre.setDefaultEdgeLabel(function(){
-    return ;
-  });
-
-  gDagre.setNode("a", {});
-  gDagre.setEdge("b", "c");
+  gDagre.setGraph({})
+    .setDefaultEdgeLabel(function(){ return ; })
+    .setNode("a", {})
+    .setEdge("b", "c");
 
   dagre.layout(gDagre);
 }
diff --git a/dagre/dagre.d.ts b/dagre/dagre.d.ts
index a9d452826..ce389f2d7 100644
--- a/dagre/dagre.d.ts
+++ b/dagre/dagre.d.ts
@@ -14,14 +14,14 @@ declare module Dagre{
         edge(id: string): any;
         nodes(): string[];
         node(id: string): any;
-        setDefaultEdgeLabel(callback: () => void): void;
-        setEdge(sourceId: string, targetId: string): void;
-        setGraph(options: { [key: string]: any }): void;
-        setNode(id: string, node: { [key: string]: any }): void;
+        setDefaultEdgeLabel(callback: () => void): Graph;
+        setEdge(sourceId: string, targetId: string): Graph;
+        setGraph(options: { [key: string]: any }): Graph;
+        setNode(id: string, node: { [key: string]: any }): Graph;
     }
 
     interface GraphLib {
-        Graph: Graph
+        Graph: Graph;
     }
 }