diff --git a/Include/Library/OcDebugLogLib.h b/Include/Library/OcDebugLogLib.h
index bfc52cd5..b240bfd8 100644
--- a/Include/Library/OcDebugLogLib.h
+++ b/Include/Library/OcDebugLogLib.h
@@ -108,13 +108,13 @@ WaitForKeyPress (
 /**
   Obtains key index from user input.
 
-  @param TimeOutSeconds  Timeout to wait for.
+  @param[in]  Time  Timeout to wait for.
 
-  @returns key index [0, OC_INPUT_MAX), OC_INPUT_ABORTED, or OC_INPUT_INVALID.
+  @returns key index [0, OC_INPUT_MAX), OC_INPUT_ABORTED, or OC_INPUT_INVALID, OC_INPUT_MAX.
 **/
 INTN
 WaitForKeyIndex (
-  UINTN  TimeOutSeconds
+  IN UINTN  Timeout
   );
 
 /**
diff --git a/Library/OcDebugLogLib/DebugHelp.c b/Library/OcDebugLogLib/DebugHelp.c
index b66738b4..7ad2b488 100644
--- a/Library/OcDebugLogLib/DebugHelp.c
+++ b/Library/OcDebugLogLib/DebugHelp.c
@@ -66,7 +66,7 @@ WaitForKeyPress (
 
 INTN
 WaitForKeyIndex (
-  UINTN  Timeout
+  IN UINTN  Timeout
   )
 {
   EFI_STATUS        Status;