Fix prompt off doesn't work