2 files changed, 4 insertions, 4 deletions
diff --git a/lib/hosts.rb b/lib/hosts.rb
index 1a0fc27..1917b93 100644
--- a/lib/hosts.rb
+++ b/lib/hosts.rb
@@ -11,8 +11,8 @@ module SSHConfig
       @hosts += morehosts.hosts
     end
 
-    def host(name)
-      @hosts << Host.new(name, &Proc.new)
+    def host(name, &blk)
+      @hosts << Host.new(name, &blk)
     end
 
     def to_s
diff --git a/lib/sshconfig.rb b/lib/sshconfig.rb
index 204a127..81ac116 100644
--- a/lib/sshconfig.rb
+++ b/lib/sshconfig.rb
@@ -7,7 +7,7 @@ module SSHConfig
 
   $hostlist = Hosts.new{}
 
-  def hosts
-    $hostlist.merge(Hosts.new(&Proc.new))
+  def hosts(&blk)
+    $hostlist.merge(Hosts.new(&blk))
   end
 end