test: Fix collect function.