diff --git a/tests/test_plugin.py b/tests/test_plugin.py
index 1e4ec80..cf93ae4 100644
--- a/tests/test_plugin.py
+++ b/tests/test_plugin.py
@@ -2,16 +2,18 @@
 # License, v. 2.0. If a copy of the MPL was not distributed with this
 # file, You can obtain one at http://mozilla.org/MPL/2.0/.
 
-from pypom import Region
+from pypom import Region, hookimpl
 
 
 def test_after_wait_for_page_to_load(page):
     log = []
 
     class Plugin:
+        @hookimpl
         def pypom_after_wait_for_page_to_load(self, page):
             log.append(1)
 
+        @hookimpl
         def pypom_after_wait_for_region_to_load(self, region):
             log.append(2)