diff --git a/src/api.c b/src/api.c
index d376815..09cd9dc 100644
--- a/src/api.c
+++ b/src/api.c
@@ -2797,7 +2797,7 @@
 	}
 
 	if (ffa_is_framework_msg(args) &&
-	    ffa_direct_msg_handle_framework_msg(args, &ret)) {
+	    ffa_direct_msg_handle_framework_msg(args, &ret, current, next)) {
 		return ret;
 	}