website cleanup

apps/website/components/profile/ProfileHeader.tsx

@@ -1,14 +1,14 @@
 'use client';
 
 import Image from 'next/image';
-import type { GetDriverOutputDTO } from '@/lib/types/generated/GetDriverOutputDTO';
+import type { DriverViewModel } from '@/lib/view-models/DriverViewModel';
 import Button from '../ui/Button';
 import DriverRatingPill from '@/components/profile/DriverRatingPill';
 import CountryFlag from '@/components/ui/CountryFlag';
 import { useServices } from '@/lib/services/ServiceProvider';
 
 interface ProfileHeaderProps {
-  driver: GetDriverOutputDTO;
+  driver: DriverViewModel;
   rating?: number | null;
   rank?: number | null;
   isOwnProfile?: boolean;
@@ -44,7 +44,7 @@ export default function ProfileHeader({
         <div>
           <div className="flex items-center gap-3 mb-2">
             <h1 className="text-3xl font-bold text-white">{driver.name}</h1>
-            <CountryFlag countryCode={driver.country} size="lg" />
+            {driver.country && <CountryFlag countryCode={driver.country} size="lg" />}
             {teamTag && (
               <span className="px-3 py-1 bg-primary-blue/20 text-primary-blue rounded-full text-sm font-medium">
                 {teamTag}