diff --git a/docs/.vitepress/theme/components/price-estimator/stores/priceEstimatorStore.ts b/docs/.vitepress/theme/components/price-estimator/stores/priceEstimatorStore.ts
index 62bfaf7b26..c5b678352b 100644
--- a/docs/.vitepress/theme/components/price-estimator/stores/priceEstimatorStore.ts
+++ b/docs/.vitepress/theme/components/price-estimator/stores/priceEstimatorStore.ts
@@ -533,7 +533,7 @@ export const priceEstimatorStore = reactive({
     const idx = lab.selectedCompute.findIndex((c) => c.id === computeId)
     if (idx === -1) return
 
-    const prices = this.getComputePriceFromCatalogue(payload.machine_type, payload.subscription, payload.gpu)
+    const prices = this.getComputePriceFromCatalogue(payload.machine_type, payload.subscription, payload.gpu, payload.gpuCount)
     lab.selectedCompute[idx] = {
       id: computeId,
       name: payload.name,
@@ -614,6 +614,7 @@ export const priceEstimatorStore = reactive({
         core_count: c.core_count,
         ram: c.ram,
         gpu: c.gpu,
+        gpu_count: c.gpu_count,
         subscription: c.subscription,
       })),
       storage: lab.selectedStorage.map((s) => ({