another fix to handle the color generation