missing changes from previous PR

2016-06-13 17:34:08 -04:00 · 2016-06-13 17:34:08 -04:00 · 8696e57498
parent 3330677b47
commit 8696e57498
1 changed files with 2 additions and 2 deletions
--- a/questionnaire/views.py
+++ b/questionnaire/views.py
@ -812,9 +812,9 @@ def _table_headers(questions):
    ql.sort(lambda x, y: numal_sort(x.number, y.number))
    columns = []
    for q in ql:
-        if q.type == 'choice-yesnocomment':
+        if q.type.startswith('choice-yesnocomment'):
            columns.extend([q.number, q.number + "-freeform"])
-        elif q.type == 'choice-freeform':
+        elif q.type.startswith('choice-freeform'):
            columns.extend([q.number, q.number + "-freeform"])
        elif q.type.startswith('choice-multiple'):
            cl = [c.value for c in q.choice_set.all()]