diff --git a/main.cpp b/main.cpp
index 72a027f..d0770c0 100644
--- a/main.cpp
+++ b/main.cpp
@@ -421,7 +421,6 @@ int main(int argc, char* argv[]) {
     Mode mode = Mode::SEQ;
     bool use_gpu = false;
     int num_generations = 0;
-    int num_threads = 1;
     bool measure = false;
 
     if (argc < 8) {
@@ -463,7 +462,7 @@ int main(int argc, char* argv[]) {
             }
         } else if (std::string(argv[i]) == "--threads") {
             if (i + 1 < argc) {
-                num_threads = std::stoi(argv[i+1]);
+                omp_set_num_threads(std::stoi(argv[i+1]));
             } else {
                 print_usage();
                 return 1;