diff --git a/shop/lib/actions/companies.ts b/shop/lib/actions/companies.ts
index 6f533de..12d9170 100644
--- a/shop/lib/actions/companies.ts
+++ b/shop/lib/actions/companies.ts
@@ -1,3 +1,4 @@
+'use server'
 import { createAdminClient } from '@/lib/supabase/admin'
 import { revalidatePath } from 'next/cache'
 
@@ -31,3 +32,11 @@ export async function deleteCompany(id: string) {
   revalidatePath('/admin/companies')
   return true
 }
+
+export async function assignCompanyToUser(userId: string, companyId: string | null) {
+  const admin = createAdminClient()
+  const { error } = await admin.from('users').update({ company_id: companyId }).eq('id', userId)
+  if (error) throw error
+  revalidatePath('/admin/users')
+  return true
+}