Message preview UI fixes