diff --git a/static/script.js b/static/script.js
index d4e4477..45fed1d 100644
--- a/static/script.js
+++ b/static/script.js
@@ -80,7 +80,7 @@ function end_edit() {
 	var datasheet_par = '<p id="datasheet-info"><i class="fa fa-circle-o-notch fa-spin fa-fw"></i></p>';
 	$('#datasheet-input').replaceWith(datasheet_par);
 
-	var newButton = '<div class="round-button"><a href="#" onclick="init_edit()"><i class="fa fa-pencil" aria-hidden="true"></i></a></div>';
+	var newButton = '<div class="round-button"><a href="#" id="edit-button"><i class="fa fa-pencil" aria-hidden="true"></i></a></div>';
 	$('.round-button').replaceWith(newButton);
 }
 
@@ -228,7 +228,7 @@ $(document).ready(function() {
               newResults += '</div>';
 
               newResults += '<div class="results-notes">';
-              if (data[i].notes != null && notes.length > 0)
+              if (data[i].notes != null && data[i].notes.length > 0)
                 newResults += '<div class="tooltip"><i class="fa fa-sicky-note"></i><span class="tooltiptext">' + data[i].notes + '</span></div>';
                 //newResults += data[i].notes;
               newResults += '</div>';