diff --git a/CUDA/fit_gpu_kernels.h b/CUDA/fit_gpu_kernels.h
index 2d3bab37f09fd020a1c176ace9dbd309082d53ad..62e99bc5c4dbd74fc56d762e51439ed486525c26 100644
--- a/CUDA/fit_gpu_kernels.h
+++ b/CUDA/fit_gpu_kernels.h
@@ -11,7 +11,9 @@ extern "C" __global__ void fit_PVM_single_kernel(	//INPUT
 							const double* 		bvecs, 
 							const double* 		bvals, 
 							const int 		nvox, 
-							const int 		nfib, 
+							const int		ndirections,
+							const int 		nfib,
+							const int		nparams, 
 							const bool 		m_include_f0, 
 							//INPUT-OUTPUT
 							double* 		params);
@@ -21,7 +23,9 @@ extern "C" __global__ void fit_PVM_single_c_kernel(	//INPUT
 							const double* 		bvecs, 
 							const double* 		bvals, 
 							const int 		nvox, 
+							const int		ndirections,
 							const int 		nfib, 
+							const int		nparams,
 							const bool		m_eval_BIC,
 							const bool 		m_include_f0,
 							const bool	 	m_return_fanning,
@@ -34,7 +38,9 @@ extern "C" __global__ void fit_PVM_multi_kernel(	//INPUT
 							const double* 		bvecs, 
 							const double* 		bvals, 
 							const int 		nvox, 
-							const int 		nfib, 		
+							const int		ndirections,
+							const int 		nfib, 	
+							const int		nparams,	
 							const bool 		m_include_f0,
 							//OUTPUT
 							double* 		params);
@@ -45,7 +51,9 @@ extern "C" __global__ void get_residuals_PVM_single_kernel(	//INPUT
 								const double* 		bvecs, 
 								const double* 		bvals, 
 								const int 		nvox, 
+								const int		ndirections,
 								const int 		nfib, 
+								const int		nparams,
 								const bool 		m_include_f0,
 								const bool* 		includes_f0,
 								//OUTPUT
@@ -57,7 +65,9 @@ extern "C" __global__ void get_residuals_PVM_single_c_kernel(	//INPUT
 								const double* 		bvecs, 
 								const double* 		bvals, 
 								const int 		nvox, 
+								const int		ndirections,
 								const int 		nfib, 
+								const int		nparams,
 								const bool 		m_include_f0,
 								const bool* 		includes_f0,
 								//OUTPUT
@@ -70,7 +80,9 @@ extern "C" __global__ void get_residuals_PVM_multi_kernel(	//INPUT
 								const double* 		bvecs, 
 								const double* 		bvals, 
 								const int 		nvox, 
+								const int		ndirections,
 								const int 		nfib, 
+								const int		nparams,
 								const bool 		m_include_f0,
 								const bool* 		includes_f0,
 								//OUTPUT