diff --git a/include/kernels_gpu.h b/include/kernels_gpu.h
index 4c0df2201a5481781bba54da15237349d2164d17..454c1df8d8c59a4ed08fa2d0040c3de779dd242b 100644
--- a/include/kernels_gpu.h
+++ b/include/kernels_gpu.h
@@ -176,10 +176,10 @@ gpu_compute_rk4_weighted_sum(double *out, const double *in, const double *k1,
 void
 gpu_compute_leapfrog_update(double *next, const double *curr, const double *soe,
     const double *tmp, const double *ie, size_t num_dofs, double dt,
-    cudaStream_t stream = 0);
+    gpuStream_t stream = 0);
 
 void
 gpu_compute_leapfrog_update(double *next, const double *curr,
     const double *tmp, const double *im, size_t num_dofs, double dt,
-    cudaStream_t stream = 0);
+    gpuStream_t stream = 0);