diff --git a/modules/document/document.model.php b/modules/document/document.model.php
index cf8a89a50..af0005e4b 100644
--- a/modules/document/document.model.php
+++ b/modules/document/document.model.php
@@ -1496,10 +1496,6 @@ class DocumentModel extends Document
 					$args->sort_index = 'extra_sort.value';
 				}
 				$query_id = 'document.getDocumentListWithExtraVars';
-				if($args->columnList && !in_array($args->sort_index, $args->columnList))
-				{
-					$args->columnList[] = $args->sort_index;
-				}
 			}
 			else
 			{
@@ -1557,6 +1553,10 @@ class DocumentModel extends Document
 			}
 			$args->columnList[$key] = 'documents.' . $column;
 		}
+		if($args->columnList && !in_array($args->sort_index, $args->columnList))
+		{
+			$args->columnList[] = $args->sort_index;
+		}
 	}
 
 	/**