corrected type check for custom_sql relations

HudsonMC16 · Oct 23, 2020 · 525f67f · 525f67f
1 parent 1cb0b69
commit 525f67f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tableau_documents/table_relations.py b/tableau_documents/table_relations.py
@@ -47,14 +47,14 @@ def _read_existing_relations(self):
 
     @property
     def main_custom_sql(self) -> str:
-        if self.main_table.get('type') == 'stored-proc':
+        if self.main_table.get('type') == 'text':
             return self.main_table.text
         else:
             raise InvalidOptionException('Data Source does not have Custom SQL defined')
 
     @main_custom_sql.setter
     def main_custom_sql(self, new_custom_sql: str):
-        if self.main_table.get('type') == 'stored-proc':
+        if self.main_table.get('type') == 'text':
             self.main_table.text = new_custom_sql
         else:
             raise InvalidOptionException('Data Source does not have Custom SQL defined')