err = dpsw_acl_add_entry(ethsw->mc_io, 0, ethsw->dpsw_handle,
                                 filter_block->acl_id, acl_entry_cfg);
 
-       dma_unmap_single(dev, acl_entry_cfg->key_iova, sizeof(cmd_buff),
+       dma_unmap_single(dev, acl_entry_cfg->key_iova,
+                        DPAA2_ETHSW_PORT_ACL_CMD_BUF_SIZE,
                         DMA_TO_DEVICE);
        if (err) {
                dev_err(dev, "dpsw_acl_add_entry() failed %d\n", err);
        err = dpsw_acl_remove_entry(ethsw->mc_io, 0, ethsw->dpsw_handle,
                                    block->acl_id, acl_entry_cfg);
 
-       dma_unmap_single(dev, acl_entry_cfg->key_iova, sizeof(cmd_buff),
-                        DMA_TO_DEVICE);
+       dma_unmap_single(dev, acl_entry_cfg->key_iova,
+                        DPAA2_ETHSW_PORT_ACL_CMD_BUF_SIZE, DMA_TO_DEVICE);
        if (err) {
                dev_err(dev, "dpsw_acl_remove_entry() failed %d\n", err);
                kfree(cmd_buff);