diff --git a/src/mod_vcard_xupdate_riak.erl b/src/mod_vcard_xupdate_riak.erl
index 129a0c6a2..242485bf2 100644
--- a/src/mod_vcard_xupdate_riak.erl
+++ b/src/mod_vcard_xupdate_riak.erl
@@ -8,6 +8,8 @@
 %%%-------------------------------------------------------------------
 -module(mod_vcard_xupdate_riak).
 
+-behaviour(mod_vcard_xupdate).
+
 %% API
 -export([init/2, import/2, add_xupdate/3, get_xupdate/2, remove_xupdate/2]).
 
diff --git a/src/mod_vcard_xupdate_sql.erl b/src/mod_vcard_xupdate_sql.erl
index 7f0079dd0..00bb29501 100644
--- a/src/mod_vcard_xupdate_sql.erl
+++ b/src/mod_vcard_xupdate_sql.erl
@@ -8,6 +8,8 @@
 %%%-------------------------------------------------------------------
 -module(mod_vcard_xupdate_sql).
 
+-behaviour(mod_vcard_xupdate).
+
 %% API
 -export([init/2, import/2, add_xupdate/3, get_xupdate/2, remove_xupdate/2,
 	 import/1, export/1]).