diff --git a/tests/ui_tools/test_boxes.py b/tests/ui_tools/test_boxes.py
index f97122eee3..302f6ab38e 100644
--- a/tests/ui_tools/test_boxes.py
+++ b/tests/ui_tools/test_boxes.py
@@ -5,7 +5,7 @@
 import pytest
 from pytest import param as case
 from pytest_mock import MockerFixture
-from urwid import Widget
+from urwid import Text, Widget
 from urwid_readline import ReadlineEdit
 
 from zulipterminal.api_types import (
@@ -1704,7 +1704,10 @@ def test_keypress_CYCLE_COMPOSE_FOCUS(
             else:
                 write_box.stream_box_view(stream_id)
         else:
-            write_box.private_box_view()
+            if message_being_edited:
+                write_box.private_box_edit_view()
+            else:
+                write_box.private_box_view()
         size = widget_size(write_box)
 
         def focus_val(x: str) -> int:
@@ -1730,6 +1733,7 @@ def focus_val(x: str) -> int:
                 expected_focus_col_name
             )
 
+    @pytest.mark.parametrize("message_being_edited", [(True), (False)])
     @pytest.mark.parametrize(
         "recipient_ids, expected_recipient_text",
         [
@@ -1749,14 +1753,19 @@ def test_private_box_recipient_editing(
         user_id_email_dict: Dict[int, str],
         recipient_ids: List[int],
         expected_recipient_text: str,
+        message_being_edited: bool,
     ) -> None:
         write_box.model.user_id_email_dict = user_id_email_dict
         write_box.model.user_dict = user_dict
         mocker.patch("urwid.connect_signal")
 
-        write_box.private_box_view(recipient_user_ids=recipient_ids)
+        if message_being_edited:
+            write_box.private_box_edit_view(recipient_user_ids=recipient_ids)
+            assert isinstance(write_box.to_write_box, Text)
+        else:
+            write_box.private_box_view(recipient_user_ids=recipient_ids)
+            assert isinstance(write_box.to_write_box, ReadlineEdit)
 
-        assert isinstance(write_box.to_write_box, ReadlineEdit)
         assert write_box.to_write_box.text == expected_recipient_text
 
     @pytest.mark.parametrize("key", keys_for_command("MARKDOWN_HELP"))