diff --git a/WordPress/src/main/java/org/wordpress/android/ui/sitecreation/previews/SitePreviewViewModel.kt b/WordPress/src/main/java/org/wordpress/android/ui/sitecreation/previews/SitePreviewViewModel.kt index 0bdd80f02a86..84f96ab53e6c 100644 --- a/WordPress/src/main/java/org/wordpress/android/ui/sitecreation/previews/SitePreviewViewModel.kt +++ b/WordPress/src/main/java/org/wordpress/android/ui/sitecreation/previews/SitePreviewViewModel.kt @@ -139,7 +139,9 @@ class SitePreviewViewModel @Inject constructor( return if (!onSiteFetched.isError) { val site = siteStore.getSiteBySiteId(remoteSiteId) if (site == null) { - updateUiState(SiteNotFoundInDbUiState) + withContext(mainDispatcher) { + updateUiState(SiteNotFoundInDbUiState) + } } site } else {