Distinct count error

uktrade · Dec 17, 2024 · 1cb293e · 1cb293e
1 parent bc60e85
commit 1cb293e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/quotas/business_rules.py b/quotas/business_rules.py
@@ -507,10 +507,10 @@ def check_QA5_equivalent_volumes(original_definition, initial_volume=None):
         "initial_volume",
     )
     if (
-        existing_initial_volumes.distinct("initial_volume").count() == 1
+        existing_initial_volumes.distinct().count() == 1
     ) and initial_volume is not None:
         return existing_initial_volumes[0]["initial_volume"] == initial_volume
-    elif existing_initial_volumes.distinct("initial_volume").count() <= 1:
+    elif existing_initial_volumes.distinct().count() <= 1:
         return True
 
     return False