diff --git a/sorting/wave_sort.cpp b/sorting/wave_sort.cpp
index e9f54a929..0470cf72e 100644
--- a/sorting/wave_sort.cpp
+++ b/sorting/wave_sort.cpp
@@ -28,9 +28,8 @@ namespace wave_sort {
  * @returns arr the wave sorted array
  */
 template <typename T>
-std::vector<T> waveSort(const std::vector<T> &in_arr) {
+std::vector<T> waveSort(const std::vector<T> &in_arr, int n) {
     std::vector<T> arr(in_arr);
-    int n = sizeof(in_arr)/sizeof(int);
 
     for (int i=0;i<n;i++) {
         arr[i] = in_arr[i];
@@ -52,7 +51,7 @@ static void test() {
     // [10, 90, 49, 2, 1, 5, 23] return [2, 1, 10, 5, 49, 23, 90]
     std::vector<int> array1 = {10, 90, 49, 2, 1, 5, 23};
     std::cout << "Test 1... ";
-    std::vector<int> arr1 = sorting::wave_sort::waveSort(array1);
+    std::vector<int> arr1 = sorting::wave_sort::waveSort(array1,7);
     std::vector<int> o1 = {2, 1, 10, 5, 49, 23, 90};
     assert(std::equal(arr1.begin(),arr1.end(),o1.begin(),o1.end()));
     std::cout << "passed" << std::endl;
@@ -60,7 +59,7 @@ static void test() {
     // [1, 3, 4, 2, 7, 8] return [2, 1, 4, 3, 8, 7]
     std::vector<int> array2 = {1, 3, 4, 2, 7, 8};
     std::cout << "Test 2... ";
-    std::vector<int> arr2 = sorting::wave_sort::waveSort(array2);
+    std::vector<int> arr2 = sorting::wave_sort::waveSort(array2,6);
     std::vector<int> o2 = {2, 1, 4, 3, 8, 7};
     assert(std::equal(arr2.begin(),arr2.end(),o2.begin(),o2.end()));
     std::cout << "passed" << std::endl;
@@ -68,7 +67,7 @@ static void test() {
     // [3, 3, 3, 3] return [3, 3, 3, 3]
     std::vector<int> array3 = {3, 3, 3, 3};
     std::cout << "Test 3... ";
-    std::vector<int> arr3 = sorting::wave_sort::waveSort(array3);
+    std::vector<int> arr3 = sorting::wave_sort::waveSort(array3,4);
     std::vector<int> o3 = {3, 3, 3, 3};
     assert(std::equal(arr3.begin(),arr3.end(),o3.begin(),o3.end()));
     std::cout << "passed" << std::endl;
@@ -76,7 +75,7 @@ static void test() {
     // [9, 4, 6, 8, 14, 3] return [4, 3, 8, 6, 14, 9]
     std::vector<int> array4 = {9, 4, 6, 8, 14, 3};
     std::cout << "Test 4... ";
-    std::vector<int> arr4 = sorting::wave_sort::waveSort(array4);
+    std::vector<int> arr4 = sorting::wave_sort::waveSort(array4,6);
     std::vector<int> o4 = {4, 3, 8, 6, 14, 9};
     assert(std::equal(arr4.begin(),arr4.end(),o4.begin(),o4.end()));
     std::cout << "passed" << std::endl;