diff --git a/dev/tests/unit/Magento/FunctionalTestFramework/Util/TestGeneratorTest.php b/dev/tests/unit/Magento/FunctionalTestFramework/Util/TestGeneratorTest.php
index 66a2919d4..800c506f1 100644
--- a/dev/tests/unit/Magento/FunctionalTestFramework/Util/TestGeneratorTest.php
+++ b/dev/tests/unit/Magento/FunctionalTestFramework/Util/TestGeneratorTest.php
@@ -12,6 +12,7 @@
 use Magento\FunctionalTestingFramework\Exceptions\TestReferenceException;
 use Magento\FunctionalTestingFramework\Filter\FilterList;
 use Magento\FunctionalTestingFramework\ObjectManager;
+use Magento\FunctionalTestingFramework\Test\Handlers\TestObjectHandler;
 use Magento\FunctionalTestingFramework\Test\Objects\ActionObject;
 use Magento\FunctionalTestingFramework\Test\Objects\TestHookObject;
 use Magento\FunctionalTestingFramework\Test\Objects\TestObject;
@@ -19,6 +20,7 @@
 use Magento\FunctionalTestingFramework\Util\GenerationErrorHandler;
 use Magento\FunctionalTestingFramework\Util\ModuleResolver;
 use Magento\FunctionalTestingFramework\Util\TestGenerator;
+use ReflectionClass;
 use ReflectionProperty;
 use tests\unit\Util\MagentoTestCase;
 use tests\unit\Util\TestLoggingUtil;
@@ -75,6 +77,8 @@ public function testEntityException(): void
 
         $testObject = new TestObject('sampleTest', ['merge123' => $actionObject], [], [], 'filename');
         $testGeneratorObject = TestGenerator::getInstance('', ['sampleTest' => $testObject]);
+        $this->mockTestObjectHandler();
+
         $testGeneratorObject->createAllTestFiles(null, []);
 
         // assert that no exception for createAllTestFiles and generation error is stored in GenerationErrorHandler
@@ -231,5 +235,25 @@ public static function tearDownAfterClass(): void
         $mftfAppConfigInstance = new ReflectionProperty(MftfApplicationConfig::class, 'MFTF_APPLICATION_CONTEXT');
         $mftfAppConfigInstance->setAccessible(true);
         $mftfAppConfigInstance->setValue(null);
+
+        $property = new ReflectionProperty(TestObjectHandler::class, 'testObjectHandler');
+        $property->setAccessible(true);
+        $property->setValue(null);
+    }
+
+    /**
+     * Mock test object handler for test.
+     */
+    private function mockTestObjectHandler(): void
+    {
+        $testObjectHandlerClass = new ReflectionClass(TestObjectHandler::class);
+        $testObjectHandlerConstructor = $testObjectHandlerClass->getConstructor();
+        $testObjectHandlerConstructor->setAccessible(true);
+        $testObjectHandler = $testObjectHandlerClass->newInstanceWithoutConstructor();
+        $testObjectHandlerConstructor->invoke($testObjectHandler);
+
+        $property = new ReflectionProperty(TestObjectHandler::class, 'testObjectHandler');
+        $property->setAccessible(true);
+        $property->setValue($testObjectHandler);
     }
 }