diff --git a/include/ByteTrack/BYTETracker.h b/include/ByteTrack/BYTETracker.h
index b9769e2..72f8a02 100644
--- a/include/ByteTrack/BYTETracker.h
+++ b/include/ByteTrack/BYTETracker.h
@@ -43,8 +43,8 @@ class BYTETracker
                           const int &cost_matrix_size_size,
                           const float &thresh,
                           std::vector<std::vector<int>> &matches,
-                          std::vector<int> &b_unmatched,
-                          std::vector<int> &a_unmatched) const;
+                          std::vector<int> &a_unmatched,
+                          std::vector<int> &b_unmatched) const;
 
     std::vector<std::vector<float>> calcIouDistance(const std::vector<STrackPtr> &a_tracks,
                                                     const std::vector<STrackPtr> &b_tracks) const;
diff --git a/src/BYTETracker.cpp b/src/BYTETracker.cpp
index a02abc1..ebbcadc 100644
--- a/src/BYTETracker.cpp
+++ b/src/BYTETracker.cpp
@@ -515,9 +515,9 @@ double byte_track::BYTETracker::execLapjv(const std::vector<std::vector<float>>
     }
 
     double **cost_ptr;
-    cost_ptr = new double *[sizeof(double *) * n];
+    cost_ptr = new double *[n];
     for (int i = 0; i < n; i++)
-        cost_ptr[i] = new double[sizeof(double) * n];
+        cost_ptr[i] = new double[n];
 
     for (int i = 0; i < n; i++)
     {
@@ -527,8 +527,8 @@ double byte_track::BYTETracker::execLapjv(const std::vector<std::vector<float>>
         }
     }
 
-    int* x_c = new int[sizeof(int) * n];
-    int *y_c = new int[sizeof(int) * n];
+    int* x_c = new int[n];
+    int *y_c = new int[n];
 
     int ret = lapjv_internal(n, cost_ptr, x_c, y_c);
     if (ret != 0)