int sg_max, u64 sg_address);
 static int aac_convert_sgraw2(struct aac_raw_io2 *rio2,
                                int pages, int nseg, int nseg_new);
+static void aac_probe_container_scsi_done(struct scsi_cmnd *scsi_cmnd);
 static int aac_send_srb_fib(struct scsi_cmnd* scsicmd);
 static int aac_send_hba_fib(struct scsi_cmnd *scsicmd);
 #ifdef AAC_DETAILED_STATUS_INFO
                struct fib *fibptr) {
        struct scsi_device *device;
 
-       if (unlikely(!scsicmd || !scsicmd->scsi_done)) {
+       if (unlikely(!scsicmd)) {
                dprintk((KERN_WARNING "aac_valid_context: scsi command corrupt\n"));
                aac_fib_complete(fibptr);
                return 0;
 
 static void aac_scsi_done(struct scsi_cmnd *scmd)
 {
-       scmd->scsi_done(scmd);
+       if (scmd->device->request_queue) {
+               /* SCSI command has been submitted by the SCSI mid-layer. */
+               scsi_done(scmd);
+       } else {
+               /* SCSI command has been submitted by aac_probe_container(). */
+               aac_probe_container_scsi_done(scmd);
+       }
 }
 
 static void get_container_name_callback(void *context, struct fib * fibptr)
 
 int aac_probe_container(struct aac_dev *dev, int cid)
 {
-       struct scsi_cmnd *scsicmd = kmalloc(sizeof(*scsicmd), GFP_KERNEL);
-       struct scsi_device *scsidev = kmalloc(sizeof(*scsidev), GFP_KERNEL);
+       struct scsi_cmnd *scsicmd = kzalloc(sizeof(*scsicmd), GFP_KERNEL);
+       struct scsi_device *scsidev = kzalloc(sizeof(*scsidev), GFP_KERNEL);
        int status;
 
        if (!scsicmd || !scsidev) {
                kfree(scsidev);
                return -ENOMEM;
        }
-       scsicmd->scsi_done = aac_probe_container_scsi_done;
 
        scsicmd->device = scsidev;
        scsidev->sdev_state = 0;