diff --git a/spec/spec_helper.cr b/spec/spec_helper.cr
index 067777f..7c718b0 100644
--- a/spec/spec_helper.cr
+++ b/spec/spec_helper.cr
@@ -13,7 +13,7 @@ def connect(connection_string)
       return DB.open(uri)
     rescue exc : DB::ConnectionRefused
       raise exc if Time.local > expiry
-      sleep(5)
+      sleep(5.seconds)
     end
   end
 end
diff --git a/src/tds/driver.cr b/src/tds/driver.cr
index 4eb4359..1426692 100644
--- a/src/tds/driver.cr
+++ b/src/tds/driver.cr
@@ -1,5 +1,5 @@
 class TDS::Driver < DB::Driver
-  class TDS::ConnectionBuilder < DB::ConnectionBuilder
+  class ConnectionBuilder < DB::ConnectionBuilder
     def initialize(@options : DB::Connection::Options, @tds_options : TDS::Connection::Options)
     end
 
@@ -10,7 +10,7 @@ class TDS::Driver < DB::Driver
 
   def connection_builder(uri : URI) : DB::ConnectionBuilder
     params = HTTP::Params.parse(uri.query || "")
-    ConnectionBuilder.new(connection_options(params), TDS::Connection::Options.from_uri(uri))
+    TDS::Driver::ConnectionBuilder.new(connection_options(params), TDS::Connection::Options.from_uri(uri))
   end
 end