diff --git a/hexrd/ui/table_selector_widget.py b/hexrd/ui/table_selector_widget.py
index f2a886e3f..4def84216 100644
--- a/hexrd/ui/table_selector_widget.py
+++ b/hexrd/ui/table_selector_widget.py
@@ -142,7 +142,7 @@ def selected_columns(self):
 class TableRowSelectorDialog(TableSelectorDialog):
     def set_options(self):
         super().set_options()
-        self.table.setSelectionBehavior(self.table.SelectRows)
+        self.table.setSelectionBehavior(QTableWidget.SelectRows)
 
         self.table.verticalHeader().hide()
         self.table.horizontalHeader().setStretchLastSection(True)
@@ -156,7 +156,7 @@ def on_selection_changed(self):
 class TableSingleRowSelectorDialog(TableRowSelectorDialog):
     def set_options(self):
         super().set_options()
-        self.table.setSelectionMode(self.table.SingleSelection)
+        self.table.setSelectionMode(QTableWidget.SingleSelection)
 
     @property
     def selected_row(self):