diff --git a/examples/b2b/app/components/Header.tsx b/examples/b2b/app/components/Header.tsx
index 410c0645bc..003e3e8d00 100644
--- a/examples/b2b/app/components/Header.tsx
+++ b/examples/b2b/app/components/Header.tsx
@@ -4,6 +4,10 @@ import {useState, Suspense} from 'react';
 import type {HeaderQuery} from 'storefrontapi.generated';
 import type {LayoutProps} from './Layout';
 import {useRootLoaderData} from '~/root';
+import type {
+  CompanyLocation,
+  CompanyLocationConnection,
+} from '@shopify/hydrogen-react/customer-account-api-types';
 
 /***********************************************/
 /**********  EXAMPLE UPDATE STARTS  ************/
@@ -156,7 +160,7 @@ function CartToggle({cart}: Pick<HeaderProps, 'cart'>) {
 /**********  EXAMPLE UPDATE STARTS  ************/
 function LocationDropdown({company}: Pick<HeaderProps, 'company'>) {
   const locations = company?.locations?.edges
-    ? company.locations.edges.map((location) => {
+    ? company.locations.edges.map((location: CompanyLocationConnection) => {
         return {...location.node};
       })
     : [];
@@ -187,7 +191,7 @@ function LocationDropdown({company}: Pick<HeaderProps, 'company'>) {
             value={selectedLocation}
             style={{marginRight: '4px'}}
           >
-            {locations.map((location) => {
+            {locations.map((location: CompanyLocation) => {
               return (
                 <option
                   defaultValue={selectedLocation}
diff --git a/examples/b2b/app/components/LocationSelector.tsx b/examples/b2b/app/components/LocationSelector.tsx
index 83519b38e8..1bfede466b 100644
--- a/examples/b2b/app/components/LocationSelector.tsx
+++ b/examples/b2b/app/components/LocationSelector.tsx
@@ -2,12 +2,13 @@ import {CartForm} from '@shopify/hydrogen';
 import type {
   Company,
   CompanyLocation,
+  CompanyLocationConnection,
 } from '@shopify/hydrogen-react/customer-account-api-types';
 
 export function LocationSelector({company}: {company: Company}) {
   const locations = company?.locations?.edges
-    ? company.locations.edges.map((loc) => {
-        return {...loc.node};
+    ? company.locations.edges.map((location: CompanyLocationConnection) => {
+        return {...location.node};
       })
     : [];
 
@@ -42,7 +43,7 @@ export function LocationSelector({company}: {company: Company}) {
       <CartForm route="/cart" action={CartForm.ACTIONS.BuyerIdentityUpdate}>
         <fieldset>
           <legend>Choose a location:</legend>
-          {locations.map((location) => {
+          {locations.map((location: CompanyLocation) => {
             return (
               <div key={location.id}>
                 <LocationItem location={location} />
diff --git a/examples/b2b/app/routes/cart.tsx b/examples/b2b/app/routes/cart.tsx
index 70d503ef46..a49a31317c 100644
--- a/examples/b2b/app/routes/cart.tsx
+++ b/examples/b2b/app/routes/cart.tsx
@@ -5,6 +5,7 @@ import {CartForm} from '@shopify/hydrogen';
 import {json, type ActionFunctionArgs} from '@shopify/remix-oxygen';
 import {CartMain} from '~/components/Cart';
 import {useRootLoaderData} from '~/root';
+import type {CartBuyerIdentityInput} from '@shopify/hydrogen/storefront-api-types';
 
 export const meta: MetaFunction = () => {
   return [{title: `Hydrogen | Cart`}];
@@ -20,7 +21,8 @@ export async function action({request, context}: ActionFunctionArgs) {
   /***********************************************/
   /**********  EXAMPLE UPDATE STARTS  ************/
 
-  const companyLocationId = inputs.companyLocationId as string; // fix this type properly
+  const companyLocationId =
+    inputs.companyLocationId as CartBuyerIdentityInput['companyLocationId'];
 
   if (companyLocationId) {
     context.customerAccount.UNSTABLE_setBuyer({companyLocationId});