fix to correct colors based on @problem provided values