// TODO: Add collapse tests