Fix collect bug