diff --git a/assets/javascripts/redmine_dmsf.js b/assets/javascripts/redmine_dmsf.js
index 67eb4abc..f4b21282 100644
--- a/assets/javascripts/redmine_dmsf.js
+++ b/assets/javascripts/redmine_dmsf.js
@@ -133,7 +133,7 @@ function dmsfExpandRows(project_id, folder_id, parentRow, url) {
       if( m && (data.indexOf(' ' +  m[1] + ' ') < 0)) {
 
         $(parentRow).removeClass('dmsf-expanded');
-        $(parentRow).find('div.row-control').removeClass('row-control');
+	$(parentRow).find('div.dmsf-row-control').removeClass('row-control dmsf-row-control');
 
         if(!$(parentRow).hasClass('dmsf-child')) {
 
@@ -156,4 +156,4 @@ function noteMandatory(mandatory) {
   if(mandatory){
     note.focus();
   }
-}
\ No newline at end of file
+}