diff --git a/apps/common/templatetags/attribute_tags.py b/apps/common/templatetags/attribute_tags.py
index 2fafff9a97..cea6f1a49c 100644
--- a/apps/common/templatetags/attribute_tags.py
+++ b/apps/common/templatetags/attribute_tags.py
@@ -12,5 +12,9 @@ def object_property(value, arg):
     return return_attrib(value, arg)
     
 @register.filter
-def get_model_list_columns(value):
-    return model_list_columns.get(type(value), [])
+def get_model_list_columns(obj):
+    for key, value in model_list_columns.items():
+        if isinstance(obj, key):
+            return value
+
+    return []