small refinement

dachad · Dec 8, 2017 · 3f2a9b7 · 3f2a9b7
1 parent 29a2ba3
commit 3f2a9b7
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/cmd/tcpgoon.go b/cmd/tcpgoon.go
@@ -63,11 +63,12 @@ func validateRequiredArgs(params *tcpgoonParams, args []string) error {
 		return errors.New("Number of required parameters doesn't match")
 	}
 	params.hostPtr = args[0]
-	if port, err := strconv.Atoi(args[1]); err != nil && port <= 0 {
+	port, err := strconv.Atoi(args[1])
+	if err != nil && port <= 0 {
 		return errors.New("Port argument is not a valid integer")
-	} else {
-		params.portPtr = port
 	}
+	params.portPtr = port
+
 	return nil
 }