Fixed legend colors.

Line 8 is the important change.