Fixed issue with UI state after stopping media