diff --git a/src/velociraptor_interface.c b/src/velociraptor_interface.c
index 218beb1976fbf813e5fc6381fa68a4e74f483d99..6b5c1aac01ed6e97fe9ffe39b0bd6b216a1868d8 100644
--- a/src/velociraptor_interface.c
+++ b/src/velociraptor_interface.c
@@ -151,8 +151,8 @@ void velociraptor_invoke(struct engine *e) {
 
     struct space *s = e->s;
     struct gpart *gparts = s->gparts;
-    const int nr_gparts = s->nr_gparts;
-    const int nr_hydro_parts = s->nr_parts;
+    const size_t nr_gparts = s->nr_gparts;
+    const size_t nr_hydro_parts = s->nr_parts;
     const int nr_cells = s->nr_cells;
     int *cell_node_ids;
 
diff --git a/src/velociraptor_interface.h b/src/velociraptor_interface.h
index cc4002dbdcb2147a85bdb06d324e1ef44719f845..9ff2781375e754d15ed9a5608d74ea38c9b6ac0c 100644
--- a/src/velociraptor_interface.h
+++ b/src/velociraptor_interface.h
@@ -69,7 +69,7 @@ struct siminfo {
 
 /* VELOCIraptor interface. */
 int InitVelociraptor(char* config_name, char* output_name, struct cosmoinfo cosmo_info, struct unitinfo unit_info, struct siminfo sim_info);
-int InvokeVelociraptor(const int num_gravity_parts, const int num_hydro_parts, struct gpart *gravity_parts, const int *cell_node_ids, char* output_name);
+int InvokeVelociraptor(const size_t num_gravity_parts, const size_t num_hydro_parts, struct gpart *gravity_parts, const int *cell_node_ids, char* output_name);
 
 /* VELOCIraptor wrapper functions. */
 void velociraptor_init(struct engine *e);