diff --git a/sc2/src/sc2code/tactrans.h b/sc2/src/sc2code/tactrans.h
index 4ec8b2492..84fbcdc71 100644
--- a/sc2/src/sc2code/tactrans.h
+++ b/sc2/src/sc2code/tactrans.h
@@ -19,9 +19,11 @@
 #ifndef _TACTRANS_H
 #define _TACTRANS_H
 
-bool battleEndReadyHuman (COUNT side);
-bool battleEndReadyComputer (COUNT side);
-bool battleEndReadyNetwork (COUNT side);
+bool battleEndReadyHuman (HumanInputContext *context);
+bool battleEndReadyComputer (ComputerInputContext *context);
+#ifdef NETPLAY
+bool battleEndReadyNetwork (NetworkInputContext *context);
+#endif
 
 #endif  /* _TACTRANS_H */