diff --git a/zerver/tests/test_message_fetch.py b/zerver/tests/test_message_fetch.py
index 3734302865..fae6815e87 100644
--- a/zerver/tests/test_message_fetch.py
+++ b/zerver/tests/test_message_fetch.py
@@ -81,7 +81,7 @@ def get_recipient_id_for_stream_name(realm: Realm, stream_name: str) -> Optional
     return stream.recipient.id if stream.recipient is not None else None
 
 
-def mute_stream(realm: Realm, user_profile: str, stream_name: str) -> None:
+def mute_stream(realm: Realm, user_profile: UserProfile, stream_name: str) -> None:
     stream = get_stream(stream_name, realm)
     recipient = stream.recipient
     subscription = Subscription.objects.get(recipient=recipient, user_profile=user_profile)
diff --git a/zerver/tests/test_message_flags.py b/zerver/tests/test_message_flags.py
index c46a336dee..1d4fb2d125 100644
--- a/zerver/tests/test_message_flags.py
+++ b/zerver/tests/test_message_flags.py
@@ -1408,7 +1408,7 @@ class PersonalMessagesFlagTest(ZulipTestCase):
 
 
 class MarkUnreadTest(ZulipTestCase):
-    def mute_stream(self, stream_name: str, user: int) -> None:
+    def mute_stream(self, stream_name: str, user: UserProfile) -> None:
         realm = get_realm("zulip")
         stream = get_stream(stream_name, realm)
         recipient = stream.recipient