Fix tests

ambitioninc · Apr 21, 2017 · aed5870 · aed5870
1 parent aac80b2
commit aed5870
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/querybuilder/query.py b/querybuilder/query.py
@@ -1750,7 +1750,7 @@ def upsert(self, rows, unique_fields, update_fields, return_rows=False, return_m
                 unique_fields,
                 update_fields,
                 auto_field_name=auto_field_name,
-                return_rows=return_models
+                return_rows=return_rows or return_models
             )
 
             # get the cursor to execute the query
@@ -1769,7 +1769,7 @@ def upsert(self, rows, unique_fields, update_fields, return_rows=False, return_m
                 update_fields,
                 auto_field_name=auto_field_name,
                 only_insert=True,
-                return_rows=return_models
+                return_rows=return_rows or return_models
             )
 
             # get the cursor to execute the query