diff --git a/tests/components/test_nuheat.py b/tests/components/test_nuheat.py
index 6b486c6afcc..91a8b326bf9 100644
--- a/tests/components/test_nuheat.py
+++ b/tests/components/test_nuheat.py
@@ -36,7 +36,9 @@ class TestNuHeat(unittest.TestCase):
         mocked_nuheat.NuHeat.assert_called_with("warm", "feet")
         self.assertIn(nuheat.DOMAIN, self.hass.data)
         self.assertEquals(2, len(self.hass.data[nuheat.DOMAIN]))
-        self.assertEquals(self.hass.data[nuheat.DOMAIN][0], "thermostat123")
+        self.assertIsInstance(
+            self.hass.data[nuheat.DOMAIN][0], type(mocked_nuheat.NuHeat())
+        )
         self.assertEquals(self.hass.data[nuheat.DOMAIN][1], "thermostat123")
 
         mocked_load.assert_called_with(