diff --git a/src/partition/partitioner.cpp b/src/partition/partitioner.cpp
index 4706ce7eba9..8719f56a6c4 100644
--- a/src/partition/partitioner.cpp
+++ b/src/partition/partitioner.cpp
@@ -14,8 +14,6 @@
 
 #include <boost/assert.hpp>
 
-#include <tbb/task_scheduler_init.h>
-
 #include "util/geojson_debug_logger.hpp"
 #include "util/geojson_debug_policies.hpp"
 #include "util/json_container.hpp"
diff --git a/src/tools/partition.cpp b/src/tools/partition.cpp
index 8434a2f6b8b..eecc674f801 100644
--- a/src/tools/partition.cpp
+++ b/src/tools/partition.cpp
@@ -153,6 +153,8 @@ int main(int argc, char *argv[]) try
         return EXIT_FAILURE;
     }
 
+    tbb::task_scheduler_init init(partition_config.requested_num_threads);
+
     auto exitcode = partition::Partitioner().Run(partition_config);
 
     util::DumpMemoryStats();