diff --git a/hw/misc/vfio.c b/hw/misc/vfio.c
index 6c36c8b687ef691e89a1a48b9007e6bb0bbece43..51844cf70d5daaab1afb191994c2eec92b1eca93 100644
--- a/hw/misc/vfio.c
+++ b/hw/misc/vfio.c
@@ -3713,7 +3713,7 @@ static int vfio_connect_container(VFIOGroup *group, AddressSpace *as)
         container->iommu_data.release = vfio_listener_release;
 
         memory_listener_register(&container->iommu_data.type1.listener,
-                                 &address_space_memory);
+                                 container->space->as);
 
         if (container->iommu_data.type1.error) {
             ret = container->iommu_data.type1.error;