diff --git a/samples/gpio-pci-idio-16.c b/samples/gpio-pci-idio-16.c
index e88803fe..6c4e99b8 100644
--- a/samples/gpio-pci-idio-16.c
+++ b/samples/gpio-pci-idio-16.c
@@ -182,7 +182,7 @@ main(int argc, char *argv[])
     }
 
     if (enable_migr) {
-        ret = vfu_setup_device_migration_callbacks(vfu_ctx, 0, &migr_callbacks);
+        ret = vfu_setup_device_migration_callbacks(vfu_ctx, &migr_callbacks);
         if (ret < 0) {
             err(EXIT_FAILURE, "failed to setup device migration");
         }
diff --git a/samples/server.c b/samples/server.c
index 191f5ad8..29bf3215 100644
--- a/samples/server.c
+++ b/samples/server.c
@@ -509,7 +509,7 @@ int main(int argc, char *argv[])
         err(EXIT_FAILURE, "failed to setup BAR1 region");
     }
 
-    ret = vfu_setup_device_migration_callbacks(vfu_ctx, 0, &migr_callbacks);
+    ret = vfu_setup_device_migration_callbacks(vfu_ctx, &migr_callbacks);
     
     if (ret < 0) {
         err(EXIT_FAILURE, "failed to setup device migration");