diff --git a/lms/templates/textannotation.html b/lms/templates/textannotation.html
index adb39c1189..01d94bf054 100644
--- a/lms/templates/textannotation.html
+++ b/lms/templates/textannotation.html
@@ -54,6 +54,8 @@
 		uri = window.location.protocol;
 		for (var index = 0; index <= 9; index += 1) uri += lms_location.split("/")[index]+"/"; //Get the unit url
 	}
+    var unit_id = $('#sequence-list').find('.active').attr("data-element");
+    uri += unit_id;
 	var pagination = 100,
 		is_staff = !('${user.is_staff}'=='False'),
         options = {
diff --git a/lms/templates/videoannotation.html b/lms/templates/videoannotation.html
index 64c5db4e5d..82a820a79a 100644
--- a/lms/templates/videoannotation.html
+++ b/lms/templates/videoannotation.html
@@ -57,7 +57,8 @@
 		uri = window.location.protocol;
 		for (var index = 0; index <= 9; index += 1) uri += lms_location.split("/")[index]+"/"; //Get the unit url
 	}
-	
+	var unit_id = $('#sequence-list').find('.active').attr("data-element");
+    uri += unit_id;
 	var pagination = 100,
 	 is_staff = !('${user.is_staff}'=='False'),
         options = {