diff --git a/backend/app/plugins/wiaas/tests/unit-tests/pricing/test-wiaas-package-pricing.php b/backend/app/plugins/wiaas/tests/unit-tests/pricing/test-wiaas-package-pricing.php
index 648b2dc..ab5c693 100644
--- a/backend/app/plugins/wiaas/tests/unit-tests/pricing/test-wiaas-package-pricing.php
+++ b/backend/app/plugins/wiaas/tests/unit-tests/pricing/test-wiaas-package-pricing.php
@@ -29,8 +29,9 @@ class Wiaas_Package_Pricing_Test extends Wiaas_Unit_Test_Case {
 		);
 
 		$commision = 60;
+		$cost_margin = 0;
 
-		Wiaas_Package_Pricing::set_package_prices($package, $pricing_rules, $commision);
+		Wiaas_Package_Pricing::set_package_prices($package, $pricing_rules, $commision, $cost_margin);
 
 		$configured_prices = Wiaas_Package_Pricing::get_package_prices($package);
 
diff --git a/backend/app/plugins/wiaas/tests/unit-tests/test-wiaas-pricing.php b/backend/app/plugins/wiaas/tests/unit-tests/test-wiaas-pricing.php
index 5d62fdd..83ad6d0 100644
--- a/backend/app/plugins/wiaas/tests/unit-tests/test-wiaas-pricing.php
+++ b/backend/app/plugins/wiaas/tests/unit-tests/test-wiaas-pricing.php
@@ -31,7 +31,8 @@ class Wiaas_Pricing_Test extends Wiaas_Unit_Test_Case {
 			)
 		);
 		$commision = 50;
-		Wiaas_Package_Pricing::set_package_prices($package, $pricing_rules, $commision);
+		$cost_margin = 0;
+		Wiaas_Package_Pricing::set_package_prices($package, $pricing_rules, $commision, $cost_margin);
 
 		$expected_prices = array(
 			'purchase' => array(
@@ -229,7 +230,8 @@ class Wiaas_Pricing_Test extends Wiaas_Unit_Test_Case {
 			)
 		);
 		$commision = 50;
-		Wiaas_Package_Pricing::set_package_prices($addon_package, $pricing_rules, $commision);
+		$cost_margin = 0;
+		Wiaas_Package_Pricing::set_package_prices($addon_package, $pricing_rules, $commision, $cost_margin);
 
 		Wiaas_Package_Addon::set_package_addons($package, array($addon_package->get_id()));
 
@@ -306,7 +308,8 @@ class Wiaas_Pricing_Test extends Wiaas_Unit_Test_Case {
 			)
 		);
 		$commision = 50;
-		Wiaas_Package_Pricing::set_package_prices($option_package, $pricing_rules, $commision);
+		$cost_margin = 0;
+		Wiaas_Package_Pricing::set_package_prices($option_package, $pricing_rules, $commision, $cost_margin);
 
 		Wiaas_Package_Option_Groups::set_package_option_groups($package, array(
 			'id' => 'option',