diff --git a/main.cpp b/main.cpp
index 1b97be8..96277e8 100644
--- a/main.cpp
+++ b/main.cpp
@@ -21,8 +21,8 @@ int sc_main(int argc, char* argv[]) {
     // Create an array of Interposer Interface instances
     InterposerInterface* interposers[num_chiplets];
     ChipletInterface* chiplets[num_chiplets];
-    O_E_converter* oec[num_chiplets][total_pes];
-    ProcessingElement* pe[num_chiplets][total_pes];
+    O_E_converter* oec[num_chiplets][num_pes];
+    ProcessingElement* pe[num_chiplets][num_pes];
     TokenManager* tokens[num_chiplets];
 
     for (int i = 0; i < num_chiplets; ++i) {