diff --git a/templates/solutions.html b/templates/solutions.html
index 942ceb2..d3e8cab 100644
--- a/templates/solutions.html
+++ b/templates/solutions.html
@@ -9,7 +9,10 @@
jQuery.get('/solutions_table?task_id={{ task.id }}&page=' + page.toString(), function(data) {
document.getElementById('solutions').innerHTML = data;
const name = "page_num_" + page.toString();
- document.getElementById(name).className = "btn btn-dark";
+ elem = document.getElementById(name);
+ if (elem) {
+ elem.className = "btn btn-dark";
+ }
setTimeout(function() {doPoll()}, 2000);
})
}
diff --git a/templates/task.html b/templates/task.html
index adf0d6f..56d1f80 100644
--- a/templates/task.html
+++ b/templates/task.html
@@ -27,7 +27,10 @@
jQuery.get('/solutions_table?task_id={{ task.id }}&username={{ user.username }}&page=' + page.toString(), function(data) {
document.getElementById('solutions').innerHTML = data;
const name = "page_num_" + page.toString();
- document.getElementById(name).className = "btn btn-dark";
+ elem = document.getElementById(name);
+ if (elem) {
+ elem.className = "btn btn-dark";
+ }
jQuery.get('/task_runtime?task_id={{ task.id }}', function(data1) {
document.getElementById('runtime').innerHTML = data1;
setTimeout(function() {doPoll()}, 2000);