Fix missing assignment

This broke the color selection, ops!