Correction needed for use of collections.

Co-authored-by: Kevin Lin <kevinl@cs.uw.edu>