diff --git a/src/audio/mixer/mixer.c b/src/audio/mixer/mixer.c
index 703e87ecee43..1ced093327e2 100644
--- a/src/audio/mixer/mixer.c
+++ b/src/audio/mixer/mixer.c
@@ -14,7 +14,6 @@
 #include <sof/common.h>
 #include <rtos/panic.h>
 #include <sof/ipc/msg.h>
-#include <rtos/alloc.h>
 #include <rtos/init.h>
 #include <sof/lib/uuid.h>
 #include <sof/list.h>
@@ -47,7 +46,7 @@ static int mixer_init(struct processing_module *mod)
 
 	comp_dbg(dev, "mixer_init()");
 
-	md = rzalloc(SOF_MEM_FLAG_USER, sizeof(*md));
+	md = mod_zalloc(mod, sizeof(*md));
 	if (!md)
 		return -ENOMEM;
 
@@ -66,7 +65,7 @@ static int mixer_free(struct processing_module *mod)
 
 	comp_dbg(dev, "mixer_free()");
 
-	rfree(md);
+	mod_free(mod, md);
 
 	return 0;
 }
diff --git a/src/audio/mixin_mixout/mixin_mixout.c b/src/audio/mixin_mixout/mixin_mixout.c
index ec473baf6e58..08d3ebbd9030 100644
--- a/src/audio/mixin_mixout/mixin_mixout.c
+++ b/src/audio/mixin_mixout/mixin_mixout.c
@@ -13,7 +13,6 @@
 #include <rtos/panic.h>
 #include <sof/ipc/msg.h>
 #include <sof/ipc/notification_pool.h>
-#include <rtos/alloc.h>
 #include <rtos/init.h>
 #include <sof/lib/uuid.h>
 #include <sof/list.h>
@@ -141,7 +140,7 @@ static int mixin_init(struct processing_module *mod)
 
 	comp_dbg(dev, "entry");
 
-	md = rzalloc(SOF_MEM_FLAG_USER, sizeof(*md));
+	md = mod_zalloc(mod, sizeof(*md));
 	if (!md)
 		return -ENOMEM;
 
@@ -168,7 +167,7 @@ static int mixout_init(struct processing_module *mod)
 
 	comp_dbg(dev, "entry");
 
-	mo_data = rzalloc(SOF_MEM_FLAG_USER, sizeof(*mo_data));
+	mo_data = mod_zalloc(mod, sizeof(*mo_data));
 	if (!mo_data)
 		return -ENOMEM;
 
@@ -184,14 +183,14 @@ static int mixin_free(struct processing_module *mod)
 {
 	struct mixin_data *md = module_get_private_data(mod);
 
-	rfree(md);
+	mod_free(mod, md);
 
 	return 0;
 }
 
 static int mixout_free(struct processing_module *mod)
 {
-	rfree(module_get_private_data(mod));
+	mod_free(mod, module_get_private_data(mod));
 
 	return 0;
 }
diff --git a/src/audio/mux/mux.c b/src/audio/mux/mux.c
index bbcf1b544e90..d23a734e28fe 100644
--- a/src/audio/mux/mux.c
+++ b/src/audio/mux/mux.c
@@ -11,7 +11,6 @@
 #include <sof/audio/ipc-config.h>
 #include <sof/common.h>
 #include <sof/ipc/msg.h>
-#include <rtos/alloc.h>
 #include <rtos/init.h>
 #include <sof/lib/uuid.h>
 #include <sof/list.h>
@@ -94,14 +93,13 @@ static int mux_demux_common_init(struct processing_module *mod, enum sof_comp_ty
 		return -EINVAL;
 	}
 
-	cd = rzalloc(SOF_MEM_FLAG_USER,
-		     sizeof(*cd) + MUX_BLOB_STREAMS_SIZE);
+	cd = mod_zalloc(mod, sizeof(*cd) + MUX_BLOB_STREAMS_SIZE);
 	if (!cd)
 		return -ENOMEM;
 
-	cd->model_handler = comp_data_blob_handler_new(dev);
+	cd->model_handler = mod_data_blob_handler_new(mod);
 	if (!cd->model_handler) {
-		comp_err(dev, "comp_data_blob_handler_new() failed.");
+		comp_err(dev, "mod_data_blob_handler_new() failed.");
 		ret = -ENOMEM;
 		goto err;
 	}
@@ -109,7 +107,7 @@ static int mux_demux_common_init(struct processing_module *mod, enum sof_comp_ty
 	module_data->private = cd;
 	ret = comp_init_data_blob(cd->model_handler, cfg->size, cfg->init_data);
 	if (ret < 0) {
-		comp_err(dev, "comp_init_data_blob() failed.");
+		comp_err(dev, "module data blob initialization failed.");
 		goto err_init;
 	}
 
@@ -119,10 +117,10 @@ static int mux_demux_common_init(struct processing_module *mod, enum sof_comp_ty
 	return 0;
 
 err_init:
-	comp_data_blob_handler_free(cd->model_handler);
+	mod_data_blob_handler_free(mod, cd->model_handler);
 
 err:
-	rfree(cd);
+	mod_free(mod, cd);
 	return ret;
 }
 
@@ -139,8 +137,8 @@ static int mux_free(struct processing_module *mod)
 
 	comp_dbg(mod->dev, "mux_free()");
 
-	comp_data_blob_handler_free(cd->model_handler);
-	rfree(cd);
+	mod_data_blob_handler_free(mod, cd->model_handler);
+	mod_free(mod, cd);
 	return 0;
 }