diff --git a/wizard-work/beam/test_kmp.py b/wizard-work/beam/test_kmp.py
index 6450095..deef00e 100644
--- a/wizard-work/beam/test_kmp.py
+++ b/wizard-work/beam/test_kmp.py
@@ -31,6 +31,7 @@ class TestKMP(unittest.TestCase):
 
     def test_count(self):
         self.assertEqual(search_count("ababab", "ab"), 3)
+        self.assertEqual(search_count("aaaa", "aa"), 3)
 
     def test_unicode(self):
         self.assertEqual(search("cafécafé", "café"), [0,4])