diff --git a/lib/shared/edit-messages-utils.js b/lib/shared/edit-messages-utils.js
--- a/lib/shared/edit-messages-utils.js
+++ b/lib/shared/edit-messages-utils.js
@@ -21,14 +21,15 @@
 } from '../utils/action-utils.js';
 import { useSelector } from '../utils/redux-utils.js';
 
-function useEditMessage(
+function useEditMessage(): (
   messageID?: string,
-): (newText: string) => Promise<SendEditMessageResult> {
+  newText: string,
+) => Promise<SendEditMessageResult> {
   const callEditMessage = useServerCall(sendEditMessage);
   const dispatchActionPromise = useDispatchActionPromise();
 
   return React.useCallback(
-    newText => {
+    (messageID, newText) => {
       invariant(messageID, 'messageID should be set!');
 
       const editMessagePromise = (async () => {
@@ -45,7 +46,7 @@
 
       return editMessagePromise;
     },
-    [messageID, dispatchActionPromise, callEditMessage],
+    [dispatchActionPromise, callEditMessage],
   );
 }