Fix doc typo and add additional if condition

Signed-off-by: Ankush Agarwal <[email protected]>
alexmavr · Dec 28, 2014 · 484f81e · 484f81e
1 parent 8296502
commit 484f81e
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/state/state.go b/state/state.go
@@ -1,6 +1,6 @@
 package state
 
-// State represents the state of a hosts
+// State represents the state of a host
 type State int
 
 const (
@@ -25,9 +25,11 @@ var states = []string{
 	"Error",
 }
 
+// Given a State type, returns its string representation
 func (s State) String() string {
-	if int(s) < len(states) {
+	if int(s) >= 0 && int(s) < len(states) {
 		return states[s]
+	} else {
+		return ""
 	}
-	return ""
 }