diff --git a/views_table_filter.module b/views_table_filter.module
index a278d78..833f124 100644
--- a/views_table_filter.module
+++ b/views_table_filter.module
@@ -32,10 +32,8 @@ function views_table_filter_form_views_ui_edit_display_form_alter(&$form, &$form
 
   $labels = $view->display_handler->get_field_labels();
   $columns = array('_none' => t('No'));
-  foreach ($style->options['columns'] as $field => $column) {
-    if ($field === $column) {
-      $columns[$field] = !empty($labels[$field]) ? $labels[$field] : $field;
-    }
+  foreach ($view->display_handler->get_option('fields') as $field => $info) {
+    $columns[$field] = !empty($info['label']) ? $info['label'] : $field;
   }
   if (count($columns) < 2) {
     return;