diff --git a/src/radix_sort_par3.cu b/src/radix_sort_par3.cu
index 3228e5d8d7d1beac9d344bae89f0dca53f537320..5155392a69341a34bcf2560c24603c04f7755d3e 100644
--- a/src/radix_sort_par3.cu
+++ b/src/radix_sort_par3.cu
@@ -30,13 +30,11 @@ void countSort(int arr[], int n, int exp)
     int d_count[10] = {0}; 
     int h_count[10] = {0};
     cudaMalloc((void**)&d_arr,sizeof(int)*n);
-    //cudaMalloc((void**)&count,sizeof(int)*10);
     cudaMemcpy(d_arr, arr, sizeof(int)*n,cudaMemcpyHostToDevice);
-    //cudaMemset(d_count,0,sizeof(int)*10);
   
     // Store count of occurrences in count[] 
     storeCount<<<1,32>>>(d_count,d_arr,n,exp); 
-    cudaMemcpy(h_count, d_count, sizeof(int)*10,cudaMemcpyDeviceToHost);
+    cudaMemcpy(h_count, d_count, sizeof(int)*n,cudaMemcpyDeviceToHost);
     // Change count[i] so that count[i] now contains actual 
     //  position of this digit in output[] 
     for (i = 1; i < 10; i++)