diff --git a/tests/gem_cpu_reloc.c b/tests/gem_cpu_reloc.c
index 2ab1d09e..773b15a5 100644
--- a/tests/gem_cpu_reloc.c
+++ b/tests/gem_cpu_reloc.c
@@ -76,8 +76,8 @@ static void copy(int fd, uint32_t batch, uint32_t src, uint32_t dst)
 	gem_reloc[0].presumed_offset = 0;
 
 	gem_reloc[1].offset = 7 * sizeof(uint32_t);
-	if (intel_get_drm_devid(fd) >= 8)
-		 gem_reloc[1].offset += sizeof(uint32_t);
+	if (intel_gen(intel_get_drm_devid(fd) >= 8))
+		gem_reloc[1].offset += sizeof(uint32_t);
 	gem_reloc[1].delta = 0;
 	gem_reloc[1].target_handle = src;
 	gem_reloc[1].read_domains = I915_GEM_DOMAIN_RENDER;