diff --git a/src/server.h b/src/server.h
index 1ca99b5ea..afd020759 100644
--- a/src/server.h
+++ b/src/server.h
@@ -1977,7 +1977,7 @@ void forceCommandPropagation(client *c, int flags);
 void preventCommandPropagation(client *c);
 void preventCommandAOF(client *c);
 void preventCommandReplication(client *c);
-int prepareForShutdown();
+int prepareForShutdown(int flags);
 #ifdef __GNUC__
 void serverLog(int level, const char *fmt, ...)
     __attribute__((format(printf, 2, 3)));