fixing a warning with fitted risk plots