Improve the way we handle the prompt.

Signed-off-by: Thiago Farina <tfarina@chromium.org>