Fixed a test that was breaking.

Fixed a bug with color balance.