From db3015e0bc139e01dd99d923b2b8358f0721a5f2 Mon Sep 17 00:00:00 2001
From: Nanne Baars <nanne.baars@owasp.org>
Date: Thu, 22 Oct 2020 17:50:49 +0200
Subject: [PATCH] When current lesson is not set do not try to remove selected
 class as there is no element selected. The current lesson is now selected
 correctly so it makes it easier to see which item in the menu is selected

---
 .../src/main/resources/static/js/goatApp/view/MenuView.js     | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/webgoat-container/src/main/resources/static/js/goatApp/view/MenuView.js b/webgoat-container/src/main/resources/static/js/goatApp/view/MenuView.js
index d71bf8c3e..b90640e3c 100644
--- a/webgoat-container/src/main/resources/static/js/goatApp/view/MenuView.js
+++ b/webgoat-container/src/main/resources/static/js/goatApp/view/MenuView.js
@@ -108,7 +108,9 @@ define(['jquery',
 		},
 
 		onLessonClick: function (elementId) {
-			$('#'+this.curLessonLinkId).removeClass('selected').parent().removeClass('selected');
+			if (this.curLessonLinkId) {
+				$('#'+this.curLessonLinkId).removeClass('selected').parent().removeClass('selected');
+			}
 			//update
 			$('#'+elementId).addClass('selected').parent().addClass('selected');
 			this.curLessonLinkId = elementId;