diff --git a/webgoat-lessons/sql-injection/src/main/resources/js/assignment10b.js b/webgoat-lessons/sql-injection/src/main/resources/js/assignment10b.js
index 42a7da6c2..fc90cd38b 100644
--- a/webgoat-lessons/sql-injection/src/main/resources/js/assignment10b.js
+++ b/webgoat-lessons/sql-injection/src/main/resources/js/assignment10b.js
@@ -1,5 +1,4 @@
 $(document).ready( () => {
-
     var editor = ace.edit("editor");
     editor.setTheme("ace/theme/monokai");
     editor.session.setMode("ace/mode/java");
@@ -14,13 +13,7 @@ $(document).ready( () => {
 });
 
 function ace_collect() {
-    let code = "";
-    $(".ace_line").each(function(i, el) {
-        var to_add = el.innerHTML;
-        if(/\/\/.*/.test(to_add)) {
-            to_add = to_add.replace(/\/\/.*/i, '');
-        }
-        code += to_add;
-    });
+    var editor = ace.edit("editor");
+    var code = editor.getValue();
     return code;
 }
\ No newline at end of file