diff --git a/src/core/gpu.h b/src/core/gpu.h
index 12314713..d5ee847d 100644
--- a/src/core/gpu.h
+++ b/src/core/gpu.h
@@ -553,7 +553,7 @@ void gpu_render_begin(gpu_stream* stream, gpu_canvas* canvas);
 void gpu_render_end(gpu_stream* stream);
 void gpu_compute_begin(gpu_stream* stream);
 void gpu_compute_end(gpu_stream* stream);
-void gpu_set_viewport(gpu_stream* stream, float viewport[4], float depthRange[4]);
+void gpu_set_viewport(gpu_stream* stream, float viewport[4], float depthRange[2]);
 void gpu_set_scissor(gpu_stream* stream, uint32_t scissor[4]);
 void gpu_push_constants(gpu_stream* stream, gpu_shader* shader, void* data, uint32_t size);
 void gpu_bind_pipeline(gpu_stream* stream, gpu_pipeline* pipeline, bool compute);
diff --git a/src/core/gpu_vk.c b/src/core/gpu_vk.c
index 7af67485..54166eee 100644
--- a/src/core/gpu_vk.c
+++ b/src/core/gpu_vk.c
@@ -1563,7 +1563,7 @@ void gpu_compute_end(gpu_stream* stream) {
   //
 }
 
-void gpu_set_viewport(gpu_stream* stream, float view[4], float depthRange[4]) {
+void gpu_set_viewport(gpu_stream* stream, float view[4], float depthRange[2]) {
   VkViewport viewport = { view[0], view[1], view[2], view[3], depthRange[0], depthRange[1] };
   vkCmdSetViewport(stream->commands, 0, 1, &viewport);
 }