diff --git a/tests/Unit/ProcessContentTaskTest.php b/tests/Unit/ProcessContentTaskTest.php
index 9b785c0..362b4e5 100644
--- a/tests/Unit/ProcessContentTaskTest.php
+++ b/tests/Unit/ProcessContentTaskTest.php
@@ -33,12 +33,9 @@ function makeTask(array $attributes = []): ContentTask
     $task->shouldReceive('markCompleted')->byDefault();
     $task->shouldReceive('markFailed')->byDefault();
 
-    foreach ($attributes as $prop => $value) {
-        $task->shouldReceive('__get')->with($prop)->andReturn($value);
-    }
-
-    // Return null for any property not explicitly configured
-    $task->shouldReceive('__get')->byDefault()->andReturn(null);
+    $task->shouldReceive('__get')->andReturnUsing(function (string $prop) use ($attributes) {
+        return $attributes[$prop] ?? null;
+    })->byDefault();
 
     return $task;
 }