AWS: error while creating AMI from Ubuntu - error initramfs/initrd

I've created a simple VM in VirtualBox and installed Ubuntu, however, I am unable to import this to AWS and generate an AMI from it.
Operating system: Ubuntu 20.04.4 LTS Kernel: Linux 5.4.0-104-generic

I've followed the steps provided according to the docs and setup role-policy.json & trust-policy.json:
https://docs.aws.amazon.com/vm-import/latest/userguide/vmie_prereqs.html#vmimport-role

https://docs.aws.amazon.com/vm-import/latest/userguide/vmimport-image-import.html

I keep running into the error:

{
    "ImportImageTasks": [
        {
            "Description": "My server VM",
            "ImportTaskId": "import-ami-xxx",
            "SnapshotDetails": [
                {
                    "DeviceName": "/dev/sde",
                    "DiskImageSize": 2362320896.0,
                    "Format": "VMDK",
                    "Status": "completed",
                    "Url": "s3://xxxx/simple-vm.ova",
                    "UserBucket": {
                        "S3Bucket": "xxx",
                        "S3Key": "simple-vm.ova"
                    }
                }
            ],
            "Status": "deleted",
            "StatusMessage": "ClientError: We were unable to read your import's initramfs/initrd to determine what drivers your import requires to run in EC2.",
            "Tags": []
        }
    ]
}

I've tried changing disk to and from .vdi and .vmdk

I've tried disabling floppy drive and update initramfs