Update to use correct color variable.