From 2c019b542e46fb3bb255adb3afda8483f183650c Mon Sep 17 00:00:00 2001
From: Matt Bell <mappum@gmail.com>
Date: Sat, 24 Jan 2015 16:27:06 -0800
Subject: [PATCH] core/commands: Fixed net diag command not outputting in
 chosen format

---
 core/commands/diag.go | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/core/commands/diag.go b/core/commands/diag.go
index 28b3673a3..2636fa591 100644
--- a/core/commands/diag.go
+++ b/core/commands/diag.go
@@ -106,6 +106,7 @@ connected peers and latencies between them.
 		switch vis {
 		case visD3:
 			res.SetOutput(bytes.NewReader(diag.GetGraphJson(info)))
+			return
 		case visDot:
 			var buf bytes.Buffer
 			w := diag.DotWriter{W: &buf}
@@ -115,6 +116,7 @@ connected peers and latencies between them.
 				return
 			}
 			res.SetOutput(io.Reader(&buf))
+			return
 		}
 
 		output, err := stdDiagOutputMarshal(standardDiagOutput(info))