Try to handle lite prompt better