[fix] Handle orientation changes correctly