diff --git a/webgoat-container/src/main/java/org/owasp/webgoat/service/RestartLessonService.java b/webgoat-container/src/main/java/org/owasp/webgoat/service/RestartLessonService.java index 5a46f90b0..4ea036996 100644 --- a/webgoat-container/src/main/java/org/owasp/webgoat/service/RestartLessonService.java +++ b/webgoat-container/src/main/java/org/owasp/webgoat/service/RestartLessonService.java @@ -61,5 +61,6 @@ public class RestartLessonService { UserTracker userTracker = userTrackerRepository.findOne(webSession.getUserName()); userTracker.reset(al); + userTrackerRepository.save(userTracker); } } diff --git a/webgoat-container/src/main/resources/static/js/goatApp/controller/LessonController.js b/webgoat-container/src/main/resources/static/js/goatApp/controller/LessonController.js index a0d8ceac0..52a341f13 100644 --- a/webgoat-container/src/main/resources/static/js/goatApp/controller/LessonController.js +++ b/webgoat-container/src/main/resources/static/js/goatApp/controller/LessonController.js @@ -191,6 +191,7 @@ define(['jquery', method:'GET' }).done(function(lessonLink) { self.loadLesson(self.name); + self.updateMenu(); }); };