bugfix: Set color order for relevant loss functions