diff --git a/lms/lms/doctype/lms_batch/lms_batch.js b/lms/lms/doctype/lms_batch/lms_batch.js
index b42ec2a4..3028e564 100644
--- a/lms/lms/doctype/lms_batch/lms_batch.js
+++ b/lms/lms/doctype/lms_batch/lms_batch.js
@@ -44,18 +44,22 @@ frappe.ui.form.on("LMS Batch", {
 					order_by: "idx",
 				},
 				callback: (data) => {
-					data.message.forEach((row) => {
-						let child = frm.add_child("timetable");
-						child.reference_doctype = row.reference_doctype;
-						child.reference_docname = row.reference_docname;
-						child.date = row.date;
-						child.start_time = row.start_time;
-						child.end_time = row.end_time;
-					});
-					frm.refresh_field("timetable");
-					frm.save();
+					add_timetable_rows(frm, data.message);
 				},
 			});
 		}
 	},
 });
+
+const add_timetable_rows = (frm, timetable) => {
+	timetable.forEach((row) => {
+		let child = frm.add_child("timetable");
+		child.reference_doctype = row.reference_doctype;
+		child.reference_docname = row.reference_docname;
+		child.date = row.date;
+		child.start_time = row.start_time;
+		child.end_time = row.end_time;
+	});
+	frm.refresh_field("timetable");
+	frm.save();
+};
diff --git a/lms/lms/doctype/lms_timetable_template/lms_timetable_template.js b/lms/lms/doctype/lms_timetable_template/lms_timetable_template.js
index d76abc7c..a9bff04e 100644
--- a/lms/lms/doctype/lms_timetable_template/lms_timetable_template.js
+++ b/lms/lms/doctype/lms_timetable_template/lms_timetable_template.js
@@ -4,12 +4,7 @@
 frappe.ui.form.on("LMS Timetable Template", {
 	refresh(frm) {
 		frm.set_query("reference_doctype", "timetable", function () {
-			let doctypes = [
-				"Course Lesson",
-				"LMS Quiz",
-				"LMS Assignment",
-				"LMS Live Class",
-			];
+			let doctypes = ["Course Lesson", "LMS Quiz", "LMS Assignment"];
 			return {
 				filters: {
 					name: ["in", doctypes],