diff --git a/ webgoat/main/project/JavaSource/org/owasp/webgoat/lessons/RoleBasedAccessControl/RoleBasedAccessControl.java b/ webgoat/main/project/JavaSource/org/owasp/webgoat/lessons/RoleBasedAccessControl/RoleBasedAccessControl.java
index ca1513f88..d16f6d4fa 100644
--- a/ webgoat/main/project/JavaSource/org/owasp/webgoat/lessons/RoleBasedAccessControl/RoleBasedAccessControl.java	
+++ b/ webgoat/main/project/JavaSource/org/owasp/webgoat/lessons/RoleBasedAccessControl/RoleBasedAccessControl.java	
@@ -226,6 +226,10 @@ public class RoleBasedAccessControl extends LessonAdapter
 	return hints;
     }
 
+    @Override
+	public int getStageCount() {
+		return 4;
+	}
 
     /**
      *  Gets the instructions attribute of the ParameterInjection object
diff --git a/ webgoat/main/project/JavaSource/org/owasp/webgoat/lessons/SQLInjection/SQLInjection.java b/ webgoat/main/project/JavaSource/org/owasp/webgoat/lessons/SQLInjection/SQLInjection.java
index c195b0d78..54251b5a3 100644
--- a/ webgoat/main/project/JavaSource/org/owasp/webgoat/lessons/SQLInjection/SQLInjection.java	
+++ b/ webgoat/main/project/JavaSource/org/owasp/webgoat/lessons/SQLInjection/SQLInjection.java	
@@ -240,6 +240,10 @@ public class SQLInjection extends LessonAdapter
 	return hints;
     }
 
+    @Override
+	public int getStageCount() {
+		return 4;
+	}
 
     /**
      *  Gets the instructions attribute of the ParameterInjection object