diff --git a/test/unit/bay.c b/test/unit/bay.c
index 56525f8..2030462 100644
--- a/test/unit/bay.c
+++ b/test/unit/bay.c
@@ -62,7 +62,7 @@ test_callback(struct bay *bay)
 		die("bay_register failed\n");
 
 	int64_t data = 0;
-	if (bay_add_cb(bay, BAY_CB_DIRTY, &chan, callback, &data) != 0)
+	if (bay_add_cb(bay, BAY_CB_DIRTY, &chan, callback, &data, 1) == NULL)
 		die("bay_add_cb failed\n");
 
 	if (data != 0)