diff --git a/erpnext/accounts/doctype/purchase_invoice/test_purchase_invoice.py b/erpnext/accounts/doctype/purchase_invoice/test_purchase_invoice.py
index 8276c1f..f2499d2 100644
--- a/erpnext/accounts/doctype/purchase_invoice/test_purchase_invoice.py
+++ b/erpnext/accounts/doctype/purchase_invoice/test_purchase_invoice.py
@@ -998,7 +998,7 @@
 		'expense_account': args.expense_account or '_Test Account Cost for Goods Sold - _TC',
 		"conversion_factor": 1.0,
 		"serial_no": args.serial_no,
-		"stock_uom": "_Test UOM",
+		"stock_uom": args.uom or "_Test UOM",
 		"cost_center": args.cost_center or "_Test Cost Center - _TC",
 		"project": args.project,
 		"rejected_warehouse": args.rejected_warehouse or "",
diff --git a/erpnext/regional/report/uae_vat_201/test_uae_vat_201.py b/erpnext/regional/report/uae_vat_201/test_uae_vat_201.py
index 648d98e..68c163d 100644
--- a/erpnext/regional/report/uae_vat_201/test_uae_vat_201.py
+++ b/erpnext/regional/report/uae_vat_201/test_uae_vat_201.py
@@ -324,12 +324,14 @@
 	pi = make_purchase_invoice(
 			company="_Test Company UAE VAT",
 			supplier = '_Test UAE Supplier',
+			supplier_warehouse = '_Test UAE VAT Supplier Warehouse - _TCUV',
 			warehouse = '_Test UAE VAT Supplier Warehouse - _TCUV',
 			currency = 'AED',
 			cost_center = 'Main - _TCUV',
 			expense_account = 'Cost of Goods Sold - _TCUV',
 			item = "_Test UAE VAT Item",
 			do_not_save=1,
+			uom = "Nos"
 		)
 	pi.append("taxes", {
 			"charge_type": "On Net Total",
@@ -341,18 +343,19 @@
 
 	pi.recoverable_standard_rated_expenses = 1
 
-	pi.insert()
 	pi.submit()
 
 	pi = make_purchase_invoice(
 			company="_Test Company UAE VAT",
 			supplier = '_Test UAE Supplier',
+			supplier_warehouse = '_Test UAE VAT Supplier Warehouse - _TCUV',
 			warehouse = '_Test UAE VAT Supplier Warehouse - _TCUV',
 			currency = 'AED',
 			cost_center = 'Main - _TCUV',
 			expense_account = 'Cost of Goods Sold - _TCUV',
 			item = "_Test UAE VAT Item",
 			do_not_save=1,
+			uom = "Nos"
 		)
 
 	pi.append("taxes", {
@@ -367,7 +370,6 @@
 
 	pi.recoverable_reverse_charge = 100
 
-	pi.insert()
 	pi.submit()