struct s626_private {
        uint8_t ai_cmd_running;         /* ai_cmd is running */
-       uint8_t ai_continuous;          /* continuous acquisition */
        int ai_sample_count;            /* number of samples to acquire */
        unsigned int ai_sample_timer;   /* time between samples in
                                         * units of the timer */
        /* end of scan occurs */
        async->events |= COMEDI_CB_EOS;
 
-       if (!devpriv->ai_continuous)
+       if (cmd->stop_src == TRIG_COUNT)
                devpriv->ai_sample_count--;
        if (devpriv->ai_sample_count <= 0) {
                devpriv->ai_cmd_running = 0;
        case TRIG_COUNT:
                /* data arrives as one packet */
                devpriv->ai_sample_count = cmd->stop_arg;
-               devpriv->ai_continuous = 0;
                break;
        case TRIG_NONE:
                /* continuous acquisition */
-               devpriv->ai_continuous = 1;
                devpriv->ai_sample_count = 1;
                break;
        }