diff --git a/test/support/custom_partitioner.ex b/test/support/custom_partitioner.ex
index bdeec0b..ebaf6e4 100644
--- a/test/support/custom_partitioner.ex
+++ b/test/support/custom_partitioner.ex
@@ -1,5 +1,5 @@
 defmodule Klife.TestCustomPartitioner do
-  @behaviour Klife.Partitioner
+  @behaviour Klife.Behaviours.Partitioner
 
   alias Klife.Record