diff --git a/src/ZM/Context/Context.php b/src/ZM/Context/Context.php
index a17d1a13..c54e705b 100644
--- a/src/ZM/Context/Context.php
+++ b/src/ZM/Context/Context.php
@@ -255,7 +255,7 @@ class Context implements ContextInterface
      * @throws InvalidArgumentException
      * @throws ZMKnownException
      * @throws WaitTimeoutException
-     * @return mixed|string
+     * @return float|int|string
      */
     public function getArgs($mode, $prompt_msg)
     {