From be7ef06a2d434d8400da4a9619abbf3d45eb321a Mon Sep 17 00:00:00 2001
From: Trianta <56975502+Trimutex@users.noreply.github.com>
Date: Fri, 8 Dec 2023 22:51:44 -0600
Subject: [PATCH] Fixed k=0 issue

---
 src/generator.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/generator.cpp b/src/generator.cpp
index 74ef1d5..2aaa27a 100644
--- a/src/generator.cpp
+++ b/src/generator.cpp
@@ -86,11 +86,11 @@ void Generator::SetArguments(const int argc, char* argv[]) {
         }
         if (tempStr == "-k") {
             setup.isPrefixSet = true;
-            setup.prefixLength = std::stoi(argv[i+1]);
+            setup.prefixLength = std::stoi(argv[i+1]) + 1;
         }
         if (tempStr == "-n") {
             setup.isOutputSet = true;
-            setup.outputLength = std::stoi(argv[i+1]);
+            setup.outputLength = std::stoi(argv[i+1]) + 1;
         }
         if (tempStr == "-h") {
             PrintUsage();