diff --git a/tests/components/imgw_pib/test_init.py b/tests/components/imgw_pib/test_init.py
index 6ee3528422d..e352c643676 100644
--- a/tests/components/imgw_pib/test_init.py
+++ b/tests/components/imgw_pib/test_init.py
@@ -55,7 +55,7 @@ async def test_remove_binary_sensor_entity(
 ) -> None:
     """Test removing a binary_sensor entity."""
     entity_id = "binary_sensor.river_name_station_name_flood_alarm"
-    await init_integration(hass, mock_config_entry)
+    mock_config_entry.add_to_hass(hass)
 
     entity_registry.async_get_or_create(
         BINARY_SENSOR_PLATFORM,
@@ -65,4 +65,7 @@ async def test_remove_binary_sensor_entity(
         config_entry=mock_config_entry,
     )
 
+    await hass.config_entries.async_setup(mock_config_entry.entry_id)
+    await hass.async_block_till_done()
+
     assert hass.states.get(entity_id) is None
diff --git a/tests/components/imgw_pib/test_sensor.py b/tests/components/imgw_pib/test_sensor.py
index aebe4d2f86e..a1920f38006 100644
--- a/tests/components/imgw_pib/test_sensor.py
+++ b/tests/components/imgw_pib/test_sensor.py
@@ -76,7 +76,7 @@ async def test_remove_entity(
 ) -> None:
     """Test removing entity."""
     entity_id = "sensor.river_name_station_name_flood_alarm_level"
-    await init_integration(hass, mock_config_entry)
+    mock_config_entry.add_to_hass(hass)
 
     entity_registry.async_get_or_create(
         SENSOR_PLATFORM,
@@ -86,4 +86,7 @@ async def test_remove_entity(
         config_entry=mock_config_entry,
     )
 
+    await hass.config_entries.async_setup(mock_config_entry.entry_id)
+    await hass.async_block_till_done()
+
     assert hass.states.get(entity_id) is None