diff --git a/test/zero_one/geni/dataset_test.clj b/test/zero_one/geni/dataset_test.clj
index ff54c18..7f3f284 100644
--- a/test/zero_one/geni/dataset_test.clj
+++ b/test/zero_one/geni/dataset_test.clj
@@ -430,7 +430,7 @@
     (-> (df-20)
         (g/repartition :Suburb :SellerG)
         g/partitions
-        count) => #(< 1 %))
+        count) => #(<= 1 %))
   (fact "able to repartition by number and columns"
     (-> (df-20)
         (g/repartition 10 :Suburb :SellerG)
diff --git a/test/zero_one/geni/sql_functions_test.clj b/test/zero_one/geni/sql_functions_test.clj
index 2999ce8..8f1cc74 100644
--- a/test/zero_one/geni/sql_functions_test.clj
+++ b/test/zero_one/geni/sql_functions_test.clj
@@ -214,7 +214,7 @@
   (-> (df-20)
       (g/cube :SellerG :Regionname)
       (g/agg (g/grouping-id :SellerG :Regionname))
-      g/first-vals) => ["Nelson" nil 1]
+      g/first-vals) => ["Biggin" "Northern Metropolitan" 0]
   (-> (df-20)
       (g/group-by :SellerG)
       (g/agg (-> (g/collect-list :Regionname) (g/as :regions)))
@@ -503,7 +503,7 @@
           (g/agg
            (g/count-distinct {:seller :SellerG
                               :suburb :Suburb}))
-          g/column-names) => ["count(SellerG AS `seller`, Suburb AS `suburb`)"])))
+          g/column-names) => ["count(SellerG AS seller, Suburb AS suburb)"])))
 
 (facts "On window functions" :slow
   (let [window  (g/window {:partition-by :SellerG :order-by :Price})]