diff --git a/Mastodon/Scene/HomeTimeline/HomeTimelineViewModel+LoadLatestState.swift b/Mastodon/Scene/HomeTimeline/HomeTimelineViewModel+LoadLatestState.swift index 932035f06d..71dd4ec0e0 100644 --- a/Mastodon/Scene/HomeTimeline/HomeTimelineViewModel+LoadLatestState.swift +++ b/Mastodon/Scene/HomeTimeline/HomeTimelineViewModel+LoadLatestState.swift @@ -51,7 +51,7 @@ extension HomeTimelineViewModel { extension HomeTimelineViewModel.LoadLatestState { class Initial: HomeTimelineViewModel.LoadLatestState { override func isValidNextState(_ stateClass: AnyClass) -> Bool { - return stateClass == Loading.self + return stateClass == Loading.self || stateClass == LoadingManually.self } } @@ -83,7 +83,7 @@ extension HomeTimelineViewModel.LoadLatestState { class Idle: HomeTimelineViewModel.LoadLatestState { override func isValidNextState(_ stateClass: AnyClass) -> Bool { - return stateClass == Loading.self + return stateClass == Loading.self || stateClass == LoadingManually.self } }