spin_lock_bh(&conn->cmd_lock);
                                list_del(&cmd->i_list);
                                spin_unlock_bh(&conn->cmd_lock);
-                               /*
-                                * Determine if a struct se_cmd is assoicated with
-                                * this struct iscsi_cmd.
-                                */
-                               if (!(cmd->se_cmd.se_cmd_flags & SCF_SE_LUN_CMD) &&
-                                   !(cmd->tmr_req))
-                                       iscsit_release_cmd(cmd);
-                               else
-                                       transport_generic_free_cmd(&cmd->se_cmd,
-                                                               1);
+
+                               iscsit_free_cmd(cmd);
                                goto get_immediate;
                        case ISTATE_SEND_NOPIN_WANT_RESPONSE:
                                spin_unlock_bh(&cmd->istate_lock);
         */
        spin_lock_bh(&conn->cmd_lock);
        list_for_each_entry_safe(cmd, cmd_tmp, &conn->conn_cmd_list, i_list) {
-               if (!(cmd->se_cmd.se_cmd_flags & SCF_SE_LUN_CMD)) {
-
-                       list_del(&cmd->i_list);
-                       spin_unlock_bh(&conn->cmd_lock);
-                       iscsit_increment_maxcmdsn(cmd, sess);
-                       /*
-                        * Special cases for active iSCSI TMR, and
-                        * transport_lookup_cmd_lun() failing from
-                        * iscsit_get_lun_for_cmd() in iscsit_handle_scsi_cmd().
-                        */
-                       if (cmd->tmr_req)
-                               transport_generic_free_cmd(&cmd->se_cmd, 1);
-                       else
-                               iscsit_release_cmd(cmd);
 
-                       spin_lock_bh(&conn->cmd_lock);
-                       continue;
-               }
                list_del(&cmd->i_list);
                spin_unlock_bh(&conn->cmd_lock);
 
                iscsit_increment_maxcmdsn(cmd, sess);
 
-               transport_generic_free_cmd(&cmd->se_cmd, 1);
+               iscsit_free_cmd(cmd);
 
                spin_lock_bh(&conn->cmd_lock);
        }
 
                        list_del(&cmd->i_list);
                        cmd->conn = NULL;
                        spin_unlock(&cr->conn_recovery_cmd_lock);
-                       if (!(cmd->se_cmd.se_cmd_flags & SCF_SE_LUN_CMD))
-                               iscsit_release_cmd(cmd);
-                       else
-                               transport_generic_free_cmd(&cmd->se_cmd, 1);
+                       iscsit_free_cmd(cmd);
                        spin_lock(&cr->conn_recovery_cmd_lock);
                }
                spin_unlock(&cr->conn_recovery_cmd_lock);
                        list_del(&cmd->i_list);
                        cmd->conn = NULL;
                        spin_unlock(&cr->conn_recovery_cmd_lock);
-                       if (!(cmd->se_cmd.se_cmd_flags & SCF_SE_LUN_CMD))
-                               iscsit_release_cmd(cmd);
-                       else
-                               transport_generic_free_cmd(&cmd->se_cmd, 1);
+                       iscsit_free_cmd(cmd);
                        spin_lock(&cr->conn_recovery_cmd_lock);
                }
                spin_unlock(&cr->conn_recovery_cmd_lock);
                iscsit_remove_cmd_from_connection_recovery(cmd, sess);
 
                spin_unlock(&cr->conn_recovery_cmd_lock);
-               if (!(cmd->se_cmd.se_cmd_flags & SCF_SE_LUN_CMD))
-                       iscsit_release_cmd(cmd);
-               else
-                       transport_generic_free_cmd(&cmd->se_cmd, 1);
+               iscsit_free_cmd(cmd);
                spin_lock(&cr->conn_recovery_cmd_lock);
        }
        spin_unlock(&cr->conn_recovery_cmd_lock);
                list_del(&cmd->i_list);
 
                spin_unlock_bh(&conn->cmd_lock);
-               if (!(cmd->se_cmd.se_cmd_flags & SCF_SE_LUN_CMD))
-                       iscsit_release_cmd(cmd);
-               else
-                       transport_generic_free_cmd(&cmd->se_cmd, 1);
+               iscsit_free_cmd(cmd);
                spin_lock_bh(&conn->cmd_lock);
        }
        spin_unlock_bh(&conn->cmd_lock);
 
                        list_del(&cmd->i_list);
                        spin_unlock_bh(&conn->cmd_lock);
-
-                       if (!(cmd->se_cmd.se_cmd_flags & SCF_SE_LUN_CMD))
-                               iscsit_release_cmd(cmd);
-                       else
-                               transport_generic_free_cmd(&cmd->se_cmd, 1);
+                       iscsit_free_cmd(cmd);
                        spin_lock_bh(&conn->cmd_lock);
                        continue;
                }
                     (cmd->cmd_sn >= conn->sess->exp_cmd_sn)) {
                        list_del(&cmd->i_list);
                        spin_unlock_bh(&conn->cmd_lock);
-
-                       if (!(cmd->se_cmd.se_cmd_flags & SCF_SE_LUN_CMD))
-                               iscsit_release_cmd(cmd);
-                       else
-                               transport_generic_free_cmd(&cmd->se_cmd, 1);
+                       iscsit_free_cmd(cmd);
                        spin_lock_bh(&conn->cmd_lock);
                        continue;
                }
 
                iscsit_free_all_datain_reqs(cmd);
 
-               if (cmd->se_cmd.se_cmd_flags & SCF_SE_LUN_CMD)
-                       transport_wait_for_tasks(&cmd->se_cmd);
+               transport_wait_for_tasks(&cmd->se_cmd);
                /*
                 * Add the struct iscsi_cmd to the connection recovery cmd list
                 */
 
        kmem_cache_free(lio_cmd_cache, cmd);
 }
 
+void iscsit_free_cmd(struct iscsi_cmd *cmd)
+{
+       /*
+        * Determine if a struct se_cmd is assoicated with
+        * this struct iscsi_cmd.
+        */
+       switch (cmd->iscsi_opcode) {
+       case ISCSI_OP_SCSI_CMD:
+       case ISCSI_OP_SCSI_TMFUNC:
+               transport_generic_free_cmd(&cmd->se_cmd, 1);
+               break;
+       default:
+               iscsit_release_cmd(cmd);
+               break;
+       }
+}
+
 int iscsit_check_session_usage_count(struct iscsi_session *sess)
 {
        spin_lock_bh(&sess->session_usage_lock);
 
 extern void iscsit_remove_cmd_from_tx_queues(struct iscsi_cmd *, struct iscsi_conn *);
 extern void iscsit_free_queue_reqs_for_conn(struct iscsi_conn *);
 extern void iscsit_release_cmd(struct iscsi_cmd *);
+extern void iscsit_free_cmd(struct iscsi_cmd *);
 extern int iscsit_check_session_usage_count(struct iscsi_session *);
 extern void iscsit_dec_session_usage_count(struct iscsi_session *);
 extern void iscsit_inc_session_usage_count(struct iscsi_session *);