diff --git a/lib/client/gnt_debug.py b/lib/client/gnt_debug.py
index 002fcc064b707fb69c42999d31c05db0fcc101e2..ffbd569c9eafc16fbdb1c98d42a2e50756d7dbc5 100644
--- a/lib/client/gnt_debug.py
+++ b/lib/client/gnt_debug.py
@@ -172,7 +172,7 @@ def TestAllocator(opts, args):
                                vcpus=opts.vcpus,
                                tags=opts.tags,
                                direction=opts.direction,
-                               allocator=opts.iallocator,
+                               iallocator=opts.iallocator,
                                evac_mode=opts.evac_mode,
                                target_groups=target_groups,
                                spindle_use=opts.spindle_use,
diff --git a/lib/cmdlib.py b/lib/cmdlib.py
index 3b099942933e11b70d986fe64b5006951889e3d3..8ff1daeb51a70f18f06b093611980a655357e1c7 100644
--- a/lib/cmdlib.py
+++ b/lib/cmdlib.py
@@ -15533,7 +15533,7 @@ class LUTestAllocator(NoHooksLU):
                                  self.op.mode, errors.ECODE_INVAL)
 
     if self.op.direction == constants.IALLOCATOR_DIR_OUT:
-      if self.op.allocator is None:
+      if self.op.iallocator is None:
         raise errors.OpPrereqError("Missing allocator name",
                                    errors.ECODE_INVAL)
     elif self.op.direction != constants.IALLOCATOR_DIR_IN:
@@ -15586,7 +15586,7 @@ class LUTestAllocator(NoHooksLU):
     if self.op.direction == constants.IALLOCATOR_DIR_IN:
       result = ial.in_text
     else:
-      ial.Run(self.op.allocator, validate=False)
+      ial.Run(self.op.iallocator, validate=False)
       result = ial.out_text
     return result
 
diff --git a/lib/opcodes.py b/lib/opcodes.py
index 9ff1a7df41847b27de06e49a4a2f4e181fa5b097..ba0940aabf50d6a08218dd015006daba72e7f359 100644
--- a/lib/opcodes.py
+++ b/lib/opcodes.py
@@ -1959,7 +1959,7 @@ class OpTestAllocator(OpCode):
       return the allocator output (direction 'out')
 
   """
-  OP_DSC_FIELD = "allocator"
+  OP_DSC_FIELD = "iallocator"
   OP_PARAMS = [
     ("direction", ht.NoDefault,
      ht.TElemOf(constants.VALID_IALLOCATOR_DIRECTIONS), None),
@@ -1973,7 +1973,7 @@ class OpTestAllocator(OpCode):
      None),
     ("disks", ht.NoDefault, ht.TMaybe(ht.TList), None),
     ("hypervisor", None, ht.TMaybeString, None),
-    ("allocator", None, ht.TMaybeString, None),
+    _PIAllocFromDesc(None),
     ("tags", ht.EmptyList, ht.TListOf(ht.TNonEmptyString), None),
     ("memory", None, ht.TMaybe(ht.TNonNegativeInt), None),
     ("vcpus", None, ht.TMaybe(ht.TNonNegativeInt), None),