diff --git a/tests/components/smarla/test_number.py b/tests/components/smarla/test_number.py
index 3589829e56c..151dc19ab2f 100644
--- a/tests/components/smarla/test_number.py
+++ b/tests/components/smarla/test_number.py
@@ -93,11 +93,15 @@ async def test_number_state_update(
 
     entity_id = entity_info["entity_id"]
 
-    assert hass.states.get(entity_id).state == "1.0"
+    state = hass.states.get(entity_id)
+    assert state is not None
+    assert state.state == "1.0"
 
     mock_number_property.get.return_value = 100
 
     await update_property_listeners(mock_number_property)
     await hass.async_block_till_done()
 
-    assert hass.states.get(entity_id).state == "100.0"
+    state = hass.states.get(entity_id)
+    assert state is not None
+    assert state.state == "100.0"
diff --git a/tests/components/smarla/test_sensor.py b/tests/components/smarla/test_sensor.py
index 196e6d2a6f0..ec5ebfb68e4 100644
--- a/tests/components/smarla/test_sensor.py
+++ b/tests/components/smarla/test_sensor.py
@@ -75,11 +75,15 @@ async def test_sensor_state_update(
 
     entity_id = entity_info["entity_id"]
 
-    assert hass.states.get(entity_id).state == "0"
+    state = hass.states.get(entity_id)
+    assert state is not None
+    assert state.state == "0"
 
     mock_sensor_property.get.return_value = entity_info["test_value"]
 
     await update_property_listeners(mock_sensor_property)
     await hass.async_block_till_done()
 
-    assert hass.states.get(entity_id).state == "1"
+    state = hass.states.get(entity_id)
+    assert state is not None
+    assert state.state == "1"
diff --git a/tests/components/smarla/test_switch.py b/tests/components/smarla/test_switch.py
index 3f83bce3819..25d6ee8b9b1 100644
--- a/tests/components/smarla/test_switch.py
+++ b/tests/components/smarla/test_switch.py
@@ -104,11 +104,15 @@ async def test_switch_state_update(
 
     entity_id = entity_info["entity_id"]
 
-    assert hass.states.get(entity_id).state == STATE_OFF
+    state = hass.states.get(entity_id)
+    assert state is not None
+    assert state.state == STATE_OFF
 
     mock_switch_property.get.return_value = True
 
     await update_property_listeners(mock_switch_property)
     await hass.async_block_till_done()
 
-    assert hass.states.get(entity_id).state == STATE_ON
+    state = hass.states.get(entity_id)
+    assert state is not None
+    assert state.state == STATE_ON