diff --git a/urbansim_templates/models/large_multinomial_logit.py b/urbansim_templates/models/large_multinomial_logit.py
index 12004cc..eb2b487 100644
--- a/urbansim_templates/models/large_multinomial_logit.py
+++ b/urbansim_templates/models/large_multinomial_logit.py
@@ -536,7 +536,11 @@ def run(self, chooser_batch_size=None, interaction_terms=None):
         self.choices = None
         
         if interaction_terms is not None:
-            uniq_intx_idx_names = set([idx for intx in interaction_terms for idx in intx.index.names])
+            if type(interaction_terms) == list:
+                uniq_intx_idx_names = set([
+                    idx for intx in interaction_terms for idx in intx.index.names])
+            else:
+                uniq_intx_idx_names = interaction_terms.index.names
             obs_extra_cols = to_list(self.chooser_size) + list(uniq_intx_idx_names)
             alts_extra_cols = to_list(self.alt_capacity) + list(uniq_intx_idx_names)