Fix post-message focus