diff --git a/packages/ui/src/based-avatar.tsx b/packages/ui/src/based-avatar.tsx
index a411146..24dc136 100644
--- a/packages/ui/src/based-avatar.tsx
+++ b/packages/ui/src/based-avatar.tsx
@@ -4,12 +4,12 @@ import type { ComponentProps } from 'react';
 import * as AvatarPrimitive from '@radix-ui/react-avatar';
 import { User } from 'lucide-react';
 
-import { AvatarImage, cn } from '@gib/ui';
+import { cn } from '@gib/ui';
 
 type BasedAvatarProps = ComponentProps<typeof AvatarPrimitive.Root> & {
   src?: string | null;
   fullName?: string | null;
-  imageProps?: Omit<ComponentProps<typeof AvatarImage>, 'data-slot'>;
+  imageProps?: Omit<ComponentProps<typeof AvatarPrimitive.Image>, 'data-slot'>;
   fallbackProps?: ComponentProps<typeof AvatarPrimitive.Fallback>;
   userIconProps?: ComponentProps<typeof User>;
 };