diff --git a/anatomy/models.py b/anatomy/models.py index 5edbcfaf..ef198e8e 100644 --- a/anatomy/models.py +++ b/anatomy/models.py @@ -151,8 +151,8 @@ class AnatomyQuestion(models.Model): return reverse("anatomy:question_detail", kwargs={"pk": self.pk}) def get_primary_answer(self): - if len(self.answers.filter(proposed=False)) > 0: - return self.answers.filter(proposed=False).first().answer + if self.answers.filter(proposed=False, status=Answer.MarkOptions.CORRECT).count() > 0: + return self.answers.filter(proposed=False, status=Answer.MarkOptions.CORRECT).first().answer else: return "None yet..." diff --git a/rapids/models.py b/rapids/models.py index ee2c906b..3f1f9d75 100644 --- a/rapids/models.py +++ b/rapids/models.py @@ -221,8 +221,8 @@ class Rapid(models.Model): def get_primary_answer(self): if self.normal: return "Normal" - elif len(self.answers.filter(proposed=False)) > 0: - return self.answers.filter(proposed=False).first().answer + elif self.answers.filter(proposed=False, status=Answer.MarkOptions.CORRECT).count() > 0: + return self.answers.filter(proposed=False, status=Answer.MarkOptions.CORRECT).first().answer else: return "None yet..."