Fix for prompt