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();