align setMessage with setCmpFlags

openwebwork · Sep 18, 2023 · 8bccd5c · 8bccd5c
1 parent b40df71
commit 8bccd5c
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/macros/parsers/parserMultiAnswer.pl b/macros/parsers/parserMultiAnswer.pl
@@ -461,10 +461,9 @@ sub perform_check {
 #  for the n-th answer blank.
 #
 sub setMessage {
-	my $self    = shift;
-	my $i       = (shift) - 1;
-	my $message = shift;
-	$self->{ans}[$i]->{ans_message} = $self->{ans}[$i]->{error_message} = $message;
+	my ($self, $i, $message) = @_;
+	die "Answer $i is not defined." unless defined($self->{ans}[ $i - 1 ]);
+	$self->{ans}[ $i - 1 ]{ans_message} = $self->{ans}[ $i - 1 ]{error_message} = $message;
 }
 
 ######################################################################