collect works better than map