Minor fix when restoring original prompt