diff --git a/erpnext/selling/doctype/sales_order/test_sales_order.py b/erpnext/selling/doctype/sales_order/test_sales_order.py
index cd669afa731..71c78bbcb29 100644
--- a/erpnext/selling/doctype/sales_order/test_sales_order.py
+++ b/erpnext/selling/doctype/sales_order/test_sales_order.py
@@ -297,7 +297,7 @@ class TestSalesOrder(unittest.TestCase):
 		frappe.db.set_value("Stock Settings", None, "auto_insert_price_list_rate_if_missing", 1)
 	
 	def test_drop_shipping(self):
-		from erpnext.selling.doctype.sales_order.sales_order import make_drop_shipment, make_delivery_note
+		from erpnext.selling.doctype.sales_order.sales_order import make_purchase_order_for_drop_shipment, make_delivery_note
 		from erpnext.stock.doctype.item.test_item import make_item
 
 		po_item = make_item("_Test Item for Drop Shipping", {"is_stock_item": 1, "is_sales_item": 1,
@@ -326,7 +326,7 @@ class TestSalesOrder(unittest.TestCase):
 		]
 
 		so = make_sales_order(item_list=so_items)
-		po = make_drop_shipment(so.name, '_Test Supplier')
+		po = make_purchase_order_for_drop_shipment(so.name, '_Test Supplier')
 		dn = make_delivery_note(so.name)
 
 		self.assertEquals(so.customer, po.customer)