diff --git a/src/main.c b/src/main.c
index b2ffccfd6ea1c9ffb7262b8281e31c4240a0f8bd..9133a19ca5de2239695930164f17a4a16a17aa12 100644
--- a/src/main.c
+++ b/src/main.c
@@ -3,7 +3,7 @@
 #include "utils/matrix.h"
 #include "utils/dijkstra.h"
 
-long int *calculate_sub_matrix(long int *matrix, int node_count, int thread_count);
+long int *calculate_sub_matrix(long int *matrix, int node_count);
 
 int main(int argc, char *argv[])
 {
@@ -23,7 +23,7 @@ int main(int argc, char *argv[])
     long int *adj_matrix = create_adj_matrix(node_count, node_count);
 
     double start_time = omp_get_wtime();
-    long int *sub_dist = calculate_sub_matrix(adj_matrix, node_count, thread_count);
+    long int *sub_dist = calculate_sub_matrix(adj_matrix, node_count);
     double time = omp_get_wtime() - start_time;
 
     printf("Elapsed Time : %f\n", time*1000000);
@@ -40,7 +40,7 @@ int main(int argc, char *argv[])
     free(adj_matrix);
 }
 
-long int *calculate_sub_matrix(long int *matrix, int node_count, int thread_count)
+long int *calculate_sub_matrix(long int *matrix, int node_count)
 {
     long int *sub_dist = (long int *)malloc(node_count * node_count * sizeof(long int));