Null check for UserHandler::userClassUpdate()

e107inc · Jan 18, 2020 · d2d0105 · d2d0105
1 parent 509b9ff
commit d2d0105
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/e107_handlers/user_handler.php b/e107_handlers/user_handler.php
@@ -955,6 +955,7 @@ public function userClassUpdate($user, $event='userfull')
 		$pref = e107::getPref();
 		$tp = e107::getParser();
 
+		$initClassStage = isset($pref['init_class_stage']) ? intval($pref['init_class_stage']) : 0;
 		$initClasses = array();
 		$doClasses = false;
 		$doProbation = false;
@@ -967,14 +968,14 @@ public function userClassUpdate($user, $event='userfull')
 				$doProbation = true;
 				break;
 			case 'userfull':
-				if(!$pref['user_reg_veri'] || (intval($pref['init_class_stage']) == 2))
+				if(!$pref['user_reg_veri'] || ($initClassStage == 2))
 				{
 					$doClasses = true;
 				}
 				$doProbation = true;
 				break;
 			case 'userpartial' :
-				if(intval($pref['init_class_stage']) === 1)
+				if($initClassStage === 1)
 				{
 					// Set initial classes if to be done on partial signup, or if selected to add them now
 					$doClasses = true;