Fix bug in prompt