diff --git a/tests/components/wemo/__init__.py b/tests/components/wemo/__init__.py
index 33bdcacd37d..68d1516aed6 100644
--- a/tests/components/wemo/__init__.py
+++ b/tests/components/wemo/__init__.py
@@ -1 +1,5 @@
 """Tests for the wemo component."""
+
+import pytest
+
+pytest.register_assert_rewrite("tests.components.wemo.entity_test_helpers")