#34: Update _get_table_to_filter

ral-facilities · Aug 22, 2019 · 3261184 · 3261184
1 parent 9e8e0e5
commit 3261184
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/common/database_helpers.py b/common/database_helpers.py
@@ -145,7 +145,7 @@ def apply_filter(self, query):
             raise BadFilterError(f" Bad operation given to where filter. operation: {self.operation}")
 
     def _get_table_to_filter(self, query):
-        if type(query) is ISISInvestigationsQuery:
+        if type(query) is ISISInvestigationsQuery or type(query) is ISISInvestigationsCountQuery:
             return INVESTIGATION
         else:
             return query.table