Fix prompting a bit