[FLINK-960] Fix CollectionDataSource bug

This closes #33.