diff --git a/homeassistant/components/smhi/__init__.py b/homeassistant/components/smhi/__init__.py
index 5b3f60f4b08..1dbfb5ecedd 100644
--- a/homeassistant/components/smhi/__init__.py
+++ b/homeassistant/components/smhi/__init__.py
@@ -40,9 +40,7 @@ async def async_migrate_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
             },
         }
 
-        if not hass.config_entries.async_update_entry(entry, data=new_data):
+        if not hass.config_entries.async_update_entry(entry, data=new_data, version=2):
             return False
 
-        entry.version = 2
-
     return True