diff --git a/main.go b/main.go
index 12bb5e3..359a492 100644
--- a/main.go
+++ b/main.go
@@ -25,10 +25,12 @@ func readArgs() {
 		inputPath = os.Args[1]
 		outPath = os.Args[2]
 		i, _ := strconv.Atoi(os.Args[3])
-		if i <= 0 {
+		finalWidth = uint(i)
+		if finalWidth <= 0 {
 			fmt.Println("Wrong input for resolution, arguments: <source_dir> <dest_dir> <width_resolution_px>")
 			os.Exit(1)
 		}
+
 	}
 }