From c277c1f742d8d984fc7ab066cd3ac513e83f0d89 Mon Sep 17 00:00:00 2001
From: ankitjavalkar
Date: Wed, 31 May 2017 19:44:43 +0530
Subject: Fix conditional logic to pass test

---
 yaksh/forms.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

(limited to 'yaksh/forms.py')

diff --git a/yaksh/forms.py b/yaksh/forms.py
index 14a3db0..2740497 100644
--- a/yaksh/forms.py
+++ b/yaksh/forms.py
@@ -246,9 +246,10 @@ class QuizForm(forms.ModelForm):
 
     def clean_prerequisite(self):
         prereq = self.cleaned_data['prerequisite']
-        if prereq and prereq.prerequisite.id == self.instance.id:
-            raise forms.ValidationError("Please set another prerequisite quiz")
-        return None
+        if prereq and prereq.prerequisite:
+            if prereq.prerequisite.id == self.instance.id:
+                raise forms.ValidationError("Please set another prerequisite quiz")
+        return prereq
 
     class Meta:
         model = Quiz
-- 
cgit