diff --git a/lms/lms/doctype/lms_enrollment/test_lms_enrollment.py b/lms/lms/doctype/lms_enrollment/test_lms_enrollment.py
index 2bef30d7..c2b9447d 100644
--- a/lms/lms/doctype/lms_enrollment/test_lms_enrollment.py
+++ b/lms/lms/doctype/lms_enrollment/test_lms_enrollment.py
@@ -10,10 +10,10 @@ from lms.lms.doctype.lms_course.test_lms_course import new_course, new_user
 
 class TestLMSEnrollment(unittest.TestCase):
 	def setUp(self):
-		frappe.db.sql("DELETE FROM `tabLMS Batch Membership`")
-		frappe.db.sql("DELETE FROM `tabLMS Batch`")
-		frappe.db.sql("delete from `tabLMS Course Mentor Mapping`")
-		frappe.db.sql("DELETE FROM `tabUser` where email like '%@test.com'")
+		frappe.db.delete("LMS Enrollment")
+		frappe.db.delete("LMS Batch Old")
+		frappe.db.delete("LMS Course Mentor Mapping")
+		frappe.db.delete("User", {"email": ("like", "%@test.com")})
 
 	def new_course_batch(self):
 		course = new_course("Test Course")
diff --git a/lms/lms/doctype/lms_exercise/test_lms_exercise.py b/lms/lms/doctype/lms_exercise/test_lms_exercise.py
index fc1f50bb..7349d632 100644
--- a/lms/lms/doctype/lms_exercise/test_lms_exercise.py
+++ b/lms/lms/doctype/lms_exercise/test_lms_exercise.py
@@ -49,6 +49,6 @@ class TestLMSExercise(unittest.TestCase):
 		assert user_submission.name == submission.name
 
 	def tearDown(self):
-		frappe.db.sql("delete from `tabLMS Batch Membership`")
-		frappe.db.sql("delete from `tabExercise Submission`")
-		frappe.db.sql("delete from `tabLMS Exercise`")
+		frappe.db.delete("LMS Enrollment")
+		frappe.db.delete("LMS Exercise Submission")
+		frappe.db.delete("LMS Exercise")