diff --git a/tests/polar_list_regression_test.cc b/tests/polar_list_regression_test.cc
index 8125537..0f53de3 100644
--- a/tests/polar_list_regression_test.cc
+++ b/tests/polar_list_regression_test.cc
@@ -51,7 +51,7 @@ int main()
 	auto temp = new simd_type[N];
 
 	double erasure_probability = 0.3;
-	int K = (1 - erasure_probability) * N;
+	int K = (1 - erasure_probability) * N + crc_aided * C;
 	double design_SNR = 10 * std::log10(-std::log(erasure_probability));
 	std::cerr << "design SNR: " << design_SNR << std::endl;
 	for (int i = 0; i < N / 32; ++i)
diff --git a/tests/polar_parity_regression_test.cc b/tests/polar_parity_regression_test.cc
index 4a5bad7..26b5f0f 100644
--- a/tests/polar_parity_regression_test.cc
+++ b/tests/polar_parity_regression_test.cc
@@ -49,7 +49,7 @@ int main()
 	auto codeword = new code_type[N];
 
 	double erasure_probability = 0.3;
-	int K = (1 - erasure_probability) * N;
+	int K = (1 - erasure_probability) * N + crc_aided * C;
 	double design_SNR = 10 * std::log10(-std::log(erasure_probability));
 	std::cerr << "design SNR: " << design_SNR << std::endl;
 	for (int i = 0; i < N / 32; ++i)