#include "qedi_fw_scsi.h"
 
 static int send_iscsi_tmf(struct qedi_conn *qedi_conn,
-                         struct iscsi_task *mtask);
+                         struct iscsi_task *mtask, struct iscsi_task *ctask);
 
 void qedi_iscsi_unmap_sg_list(struct qedi_cmd *cmd)
 {
        }
 
 send_tmf:
-       send_iscsi_tmf(qedi_conn, qedi_cmd->task);
+       send_iscsi_tmf(qedi_conn, qedi_cmd->task, ctask);
 
 clear_cleanup:
        clear_bit(QEDI_CONN_FW_CLEANUP, &qedi_conn->flags);
        clear_bit(QEDI_CONN_FW_CLEANUP, &qedi_conn->flags);
 }
 
-static int send_iscsi_tmf(struct qedi_conn *qedi_conn, struct iscsi_task *mtask)
+static int send_iscsi_tmf(struct qedi_conn *qedi_conn, struct iscsi_task *mtask,
+                         struct iscsi_task *ctask)
 {
        struct iscsi_tmf_request_hdr tmf_pdu_header;
        struct iscsi_task_params task_params;
        struct qedi_ctx *qedi = qedi_conn->qedi;
        struct e4_iscsi_task_context *fw_task_ctx;
-       struct iscsi_conn *conn = qedi_conn->cls_conn->dd_data;
-       struct iscsi_task *ctask;
        struct iscsi_tm *tmf_hdr;
        struct qedi_cmd *qedi_cmd;
        struct qedi_cmd *cmd;
 
        if ((tmf_hdr->flags & ISCSI_FLAG_TM_FUNC_MASK) ==
             ISCSI_TM_FUNC_ABORT_TASK) {
-               ctask = iscsi_itt_to_task(conn, tmf_hdr->rtt);
-               if (!ctask || !ctask->sc) {
-                       QEDI_ERR(&qedi->dbg_ctx,
-                                "Could not get reference task\n");
-                       return 0;
-               }
                cmd = (struct qedi_cmd *)ctask->dd_data;
                tmf_pdu_header.rtt =
                                qedi_set_itt(cmd->task_id,
        case ISCSI_TM_FUNC_LOGICAL_UNIT_RESET:
        case ISCSI_TM_FUNC_TARGET_WARM_RESET:
        case ISCSI_TM_FUNC_TARGET_COLD_RESET:
-               rc = send_iscsi_tmf(qedi_conn, mtask);
+               rc = send_iscsi_tmf(qedi_conn, mtask, NULL);
                break;
        default:
                QEDI_ERR(&qedi->dbg_ctx, "Invalid tmf, cid=0x%x\n",