diff --git a/backend/app/plugins/wiaas/includes/pricing/wiaas-pricing-functions.php b/backend/app/plugins/wiaas/includes/pricing/wiaas-pricing-functions.php
index a9ba343..f178532 100644
--- a/backend/app/plugins/wiaas/includes/pricing/wiaas-pricing-functions.php
+++ b/backend/app/plugins/wiaas/includes/pricing/wiaas-pricing-functions.php
@@ -70,8 +70,10 @@ function wiaas_get_package_software_procurement_cost($package) {
 	$total_cost = 0;
 
 	foreach ($bundled_items as $bundled_item) {
-		if (Wiaas_Product_Category::is_hardware($bundled_item->product)) {
-			$total_cost += Wiaas_Pricing::get_product_total_cost($bundled_item->product) * $bundled_item->get_quantity();
+		$product = $bundled_item->product;
+
+		if (Wiaas_Product_Category::get_category($product) === 'software') {
+			$total_cost += Wiaas_Pricing::get_product_total_cost($product) * $bundled_item->get_quantity();
 		}
 	}