diff --git a/lib/ioctl_wrappers.h b/lib/ioctl_wrappers.h
index 7505acbe..e575bb8d 100644
--- a/lib/ioctl_wrappers.h
+++ b/lib/ioctl_wrappers.h
@@ -89,8 +89,9 @@ struct local_i915_gem_context_param {
 	uint32_t context;
 	uint32_t size;
 	uint64_t param;
-#define LOCAL_CONTEXT_PARAM_BAN_PERIOD 0x1
-#define LOCAL_CONTEXT_PARAM_NO_ZEROMAP 0x2
+#define LOCAL_CONTEXT_PARAM_BAN_PERIOD	0x1
+#define LOCAL_CONTEXT_PARAM_NO_ZEROMAP	0x2
+#define LOCAL_CONTEXT_PARAM_GTT_SIZE	0x3
 	uint64_t value;
 };
 void gem_context_require_ban_period(int fd);
diff --git a/tests/gem_ctx_param_basic.c b/tests/gem_ctx_param_basic.c
index 45a351b8..94245ce0 100644
--- a/tests/gem_ctx_param_basic.c
+++ b/tests/gem_ctx_param_basic.c
@@ -150,7 +150,7 @@ igt_main
 	/* NOTE: This testcase intentionally tests for the next free parameter
 	 * to catch ABI extensions. Don't "fix" this testcase without adding all
 	 * the tests for the new param first. */
-	ctx_param.param = LOCAL_CONTEXT_PARAM_NO_ZEROMAP + 1;
+	ctx_param.param = LOCAL_CONTEXT_PARAM_GTT_SIZE + 1;
 
 	igt_subtest("invalid-param-get") {
 		ctx_param.context = ctx;