Set color correctly