diff --git a/test/contract/casters_test.exs b/test/contract/casters_test.exs
index 4c4f854..57eb980 100644
--- a/test/contract/casters_test.exs
+++ b/test/contract/casters_test.exs
@@ -52,7 +52,7 @@ defmodule Drops.CastersTest do
   describe ":string => :float" do
     contract do
       schema do
-        %{required(:test) => cast(:string) |> type(:float)}
+        %{required(:test) => cast(:string) |> float()}
       end
     end
 
diff --git a/test/contract/predicates_test.exs b/test/contract/predicates_test.exs
index b3b1d64..73b4593 100644
--- a/test/contract/predicates_test.exs
+++ b/test/contract/predicates_test.exs
@@ -74,7 +74,7 @@ defmodule Drops.PredicatesTest do
   describe "type?/2 with :float" do
     contract do
       schema do
-        %{required(:test) => type(:float)}
+        %{required(:test) => float()}
       end
     end