diff --git a/draggableviews.install b/draggableviews.install
index e3b2bbf..04b07b6 100644
--- a/draggableviews.install
+++ b/draggableviews.install
@@ -295,7 +295,8 @@ function draggableviews_update_7204() {
 
       // update the data
       $display_options = unserialize($query->execute()->fetchField(0));
-      $display_options['sorts']['weight']['draggableviews_setting_view'] = $sort_name;
+      $weight_key = _draggableviews_get_draggable_sort();
+      $display_options['sorts'][$weight_key]['draggableviews_setting_view'] = $sort_name;
       $display_options = serialize($display_options);
 
       // update the db