diff --git a/test/unit/support/shared/capability_helpers_context.rb b/test/unit/support/shared/capability_helpers_context.rb
index f336cc7c3..7db62d161 100644
--- a/test/unit/support/shared/capability_helpers_context.rb
+++ b/test/unit/support/shared/capability_helpers_context.rb
@@ -1,7 +1,7 @@
 shared_context "capability_helpers" do
   def detect_class(result)
     Class.new do
-      define_method(:detect?) do
+      define_method(:detect?) do |*args|
         result
       end
     end
diff --git a/test/unit/vagrant/machine_test.rb b/test/unit/vagrant/machine_test.rb
index 9cb6fe6b8..ca60f078b 100644
--- a/test/unit/vagrant/machine_test.rb
+++ b/test/unit/vagrant/machine_test.rb
@@ -225,8 +225,8 @@ describe Vagrant::Machine do
     it "should return the configured guest" do
       result = instance.guest
       result.should be_kind_of(Vagrant::Guest)
-      result.ready?.should be
-      result.chain[0][0].should == :test
+      expect(result).to be_ready
+      expect(result.capability_host_chain[0][0]).to eql(:test)
     end
   end