diff --git a/bindings/python/sample_ctl.py b/bindings/python/sample_ctl.py
index ece02b3f..5dc7d86c 100644
--- a/bindings/python/sample_ctl.py
+++ b/bindings/python/sample_ctl.py
@@ -55,7 +55,7 @@ def test_uc_ctl_tb_cache():
 
     # Now we clear cache for all TBs.
     for i in range(8):
-        uc.ctl_remove_cache(addr + i * 512)
+        uc.ctl_remove_cache(addr + i * 512, addr + 0x1000)
     
     evicted = time_emulation(uc, addr, addr + len(code))