diff --git a/tests/components/sensor/test_rest.py b/tests/components/sensor/test_rest.py
index 4abfb2d4551..1c4910927a5 100644
--- a/tests/components/sensor/test_rest.py
+++ b/tests/components/sensor/test_rest.py
@@ -7,9 +7,11 @@ from requests.exceptions import Timeout, MissingSchema, RequestException
 import requests_mock
 
 from homeassistant.bootstrap import setup_component
+import homeassistant.components.sensor as sensor
 import homeassistant.components.sensor.rest as rest
 from homeassistant.const import STATE_UNKNOWN
 from homeassistant.helpers.config_validation import template
+
 from tests.common import get_test_home_assistant, assert_setup_component
 
 
@@ -26,10 +28,9 @@ class TestRestSwitchSetup(unittest.TestCase):
 
     def test_setup_missing_config(self):
         """Test setup with configuration missing required entries."""
-        self.assertFalse(rest.setup_platform(self.hass, {
-            'platform': 'rest',
-            'resource': 'http://localhost'
-        }, None))
+        with assert_setup_component(0):
+            assert setup_component(self.hass, sensor.DOMAIN, {
+                'sensor': {'platform': 'rest'}})
 
     def test_setup_missing_schema(self):
         """Test setup with resource missing schema."""
@@ -40,7 +41,8 @@ class TestRestSwitchSetup(unittest.TestCase):
                 'method': 'GET'
             }, None)
 
-    @patch('requests.get', side_effect=requests.exceptions.ConnectionError())
+    @patch('requests.Session.send',
+           side_effect=requests.exceptions.ConnectionError())
     def test_setup_failed_connect(self, mock_req):
         """Test setup when connection error occurs."""
         self.assertFalse(rest.setup_platform(self.hass, {
@@ -48,7 +50,7 @@ class TestRestSwitchSetup(unittest.TestCase):
             'resource': 'http://localhost',
         }, None))
 
-    @patch('requests.get', side_effect=Timeout())
+    @patch('requests.Session.send', side_effect=Timeout())
     def test_setup_timeout(self, mock_req):
         """Test setup when connection timeout occurs."""
         self.assertFalse(rest.setup_platform(self.hass, {