diff --git a/spec/configs/ignite-config-base.xml b/spec/configs/ignite-config-base.xml
index f4bd067..99ad7f7 100644
--- a/spec/configs/ignite-config-base.xml
+++ b/spec/configs/ignite-config-base.xml
@@ -59,7 +59,7 @@
                     <property name="cacheMode" value="PARTITIONED"/>
                     <property name="writeSynchronizationMode" value="FULL_SYNC"/>
                     <property name="affinity">
-                        <bean class="org.apache.ignite.internal.processors.affinity.LocalAffinityFunction"/>
+                        <bean class="org.apache.ignite.compatibility.clients.JavaThinCompatibilityTest.CustomAffinity"/>
                     </property>
                 </bean>