Change to reference to collection