1 files changed, 9 insertions, 1 deletions
diff --git a/admin/element_set_unit.php b/admin/element_set_unit.php
index f4901a343..fa89db7f2 100644
--- a/admin/element_set_unit.php
+++ b/admin/element_set_unit.php
@@ -211,7 +211,15 @@ SELECT id,path,tn_ext,name,date_creation,comment,author,level,file
   }
 
   $query.= '
-  WHERE id IN ('.implode(',', $page['cat_elements_id']).')
+  WHERE id IN ('.implode(',', $page['cat_elements_id']).')';
+
+  if (is_numeric($_GET['cat']))
+  {
+    $query.= '
+    AND category_id = '.$_GET['cat'];
+  }
+
+  $query.= '
   '.$conf['order_by'].'
   LIMIT '.$page['nb_images'].' OFFSET '.$page['start'].'
 ;';