Fix collect script