From 7f6bf95aa6976da2c03ffdce9060f68c54874460 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Wed, 14 Aug 2024 15:02:52 +0200
Subject: [PATCH] Improve type hints in universal tests (#123920)

---
 tests/components/universal/test_media_player.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/components/universal/test_media_player.py b/tests/components/universal/test_media_player.py
index 187b62a93a1..69d4c8666cf 100644
--- a/tests/components/universal/test_media_player.py
+++ b/tests/components/universal/test_media_player.py
@@ -220,7 +220,7 @@ class MockMediaPlayer(media_player.MediaPlayerEntity):
 
 
 @pytest.fixture
-async def mock_states(hass):
+async def mock_states(hass: HomeAssistant) -> Mock:
     """Set mock states used in tests."""
     result = Mock()