From bde0d4ee2b9aa89b7905652ccdf1fd5506409b54 Mon Sep 17 00:00:00 2001
From: Dom Parfitt <dom.parfitt@gmail.com>
Date: Fri, 12 Oct 2018 15:49:18 +0100
Subject: [PATCH] Fixed selection extension methods

---
 types/d3-graphviz/index.d.ts | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/types/d3-graphviz/index.d.ts b/types/d3-graphviz/index.d.ts
index aab3955658..cfcde911be 100644
--- a/types/d3-graphviz/index.d.ts
+++ b/types/d3-graphviz/index.d.ts
@@ -7,13 +7,13 @@
 import { ZoomBehavior } from 'd3-zoom'
 import { Selection } from 'd3-selection'
 
-//This should be extending d3-selection, need to check it is correct
-export interface ISelection extends Selection<any, any, HTMLElement, any> {
-    graphviz(options?: IGraphvizOptions | boolean): IGraphviz;
-    selectWithoutDataPropagation(): ISelection;
+declare module 'd3-selection' {
+    interface Selection<GElement extends BaseType, Datum, PElement extends BaseType, PDatum> {
+        graphviz(options?: IGraphvizOptions | boolean): IGraphviz;
+        selectWithoutDataPropagation(): Selection<GElement, Datum, PElement, PDatum>;
+    }
 }
 
-
 /**
  * 
  * @param selector