Fixing the prompt