diff --git a/types/heapdump/heapdump-tests.ts b/types/heapdump/heapdump-tests.ts
index 49019f83fc..80306b9d32 100644
--- a/types/heapdump/heapdump-tests.ts
+++ b/types/heapdump/heapdump-tests.ts
@@ -1,9 +1,20 @@
 import * as heapdump from 'heapdump';
 
-heapdump.writeSnapshot('/tmp/myDump', (err) => {
-  if (err) {
-    console.log('Failed to dump heap: ' + err);
-  } else {
-    console.log('Successfully dumped heap!');
-  }
+let strValue = "";
+let errValue = new Error(strValue);
+let nullValue = null;
+let undefinedValue;
+
+heapdump.writeSnapshot(strValue, (err, filename) => {
+  errValue = err as Error;
+  nullValue = err as null;
+  strValue = filename as string;
+  undefinedValue = filename as undefined;
+});
+
+heapdump.writeSnapshot((err, filename) => {
+  errValue = err as Error;
+  nullValue = err as null;
+  strValue = filename as string;
+  undefinedValue = filename as undefined;
 });
diff --git a/types/heapdump/index.d.ts b/types/heapdump/index.d.ts
index d4e0477a32..1473ebb70a 100644
--- a/types/heapdump/index.d.ts
+++ b/types/heapdump/index.d.ts
@@ -2,5 +2,7 @@
 // Project: https://github.com/bnoordhuis/node-heapdump
 // Definitions by: weekens <https://github.com/weekens>
 // Definitions: https://github.com/DefinitelyTyped/DefinitelyTyped
+// TypeScript Version: 2.1
 
-export function writeSnapshot(dumpFileName: string, callback: (err?: Error) => void): void;
+export function writeSnapshot(dumpFileName?: string, callback?: (err: Error | null, filename: string | undefined) => void): void;
+export function writeSnapshot(callback: (err: Error | null, filename: string | undefined) => void): void;