ring->vaddr = (void *)(ring + 1);
        atomic_set(&ring->pin_count, 1);
 
+       ring->vma = i915_vma_alloc();
+       if (!ring->vma) {
+               kfree(ring);
+               return NULL;
+       }
+       i915_active_init(&ring->vma->active, NULL, NULL);
+
        intel_ring_update_space(ring);
 
        return ring;
 }
 
+static void mock_ring_free(struct intel_ring *ring)
+{
+       i915_active_fini(&ring->vma->active);
+       i915_vma_free(ring->vma);
+
+       kfree(ring);
+}
+
 static struct i915_request *first_request(struct mock_engine *engine)
 {
        return list_first_entry_or_null(&engine->hw_queue,
        GEM_BUG_ON(intel_context_is_pinned(ce));
 
        if (test_bit(CONTEXT_ALLOC_BIT, &ce->flags)) {
-               kfree(ce->ring);
+               mock_ring_free(ce->ring);
                mock_timeline_unpin(ce->timeline);
        }