diff --git a/src/setup.c b/src/setup.c
index 062eaa0..94fec90 100644
--- a/src/setup.c
+++ b/src/setup.c
@@ -185,7 +185,10 @@
 
 	size_t len = sizeof(*probe) + 256 * sizeof(struct io_uring_probe_op);
 	probe = malloc(len);
+	if (!probe)
+		return NULL;
 	memset(probe, 0, len);
+
 	r = io_uring_register_probe(ring, probe, 256);
 	if (r < 0)
 		goto fail;