diff --git a/antlir/antlir2/antlir2_vm/bzl/defs.bzl b/antlir/antlir2/antlir2_vm/bzl/defs.bzl
index 2ae6ccca8dc..45d795c979b 100644
--- a/antlir/antlir2/antlir2_vm/bzl/defs.bzl
+++ b/antlir/antlir2/antlir2_vm/bzl/defs.bzl
@@ -132,12 +132,12 @@ _vm_host = rule(
         "serial_index": attrs.int(default = 0, doc = "index of the serial port"),
         "use_tpm": attrs.bool(default = False, doc = "enable software TPM"),
     } | {
+        # Non-hardware parameters for the VM
         "append": attrs.option(
             attrs.string(),
             default = None,
             doc = "additional kernel command line parameter",
         ),
-        # Non-hardware parameters for the VM
         "initrd": attrs.option(
             attrs.source(),
             default = None,
diff --git a/antlir/antlir2/antlir2_vm/bzl/test.bzl b/antlir/antlir2/antlir2_vm/bzl/test.bzl
index d221a552c5f..9773e38d451 100644
--- a/antlir/antlir2/antlir2_vm/bzl/test.bzl
+++ b/antlir/antlir2/antlir2_vm/bzl/test.bzl
@@ -94,12 +94,9 @@ def _impl(ctx: AnalysisContext) -> list[Provider]:
 _vm_test = rule(
     impl = _impl,
     attrs = {
-        "test": attrs.option(
-            attrs.dep(
-                providers = [ExternalRunnerTestInfo],
-                doc = "Test target to execute inside VM",
-            ),
-            default = None,
+        "test": attrs.dep(
+            providers = [ExternalRunnerTestInfo],
+            doc = "Test target to execute inside VM",
         ),
         "test_labels": attrs.option(
             attrs.list(attrs.string(), default = []),