Check for null on colorUnit to get around Paul's issue