FAutomationEditorCommonUtils::CreateNewMap crashes the test in UE5

How do I use this workaround? I am creating simple unit tests:

IMPLEMENT_SIMPLE_AUTOMATION_TEST(TestMovement, "Blarg.Blarg.TestMovement",
								 EAutomationTestFlags::EditorContext | EAutomationTestFlags::EngineFilter)

However it is currently using the FAutomationTestBase , how do I configure the unit tests to use the custom workaround subclass?

Update: Figured it out (in case anyone else wants to use the workaround):

Use IMPLEMENT_CUSTOM_SIMPLE_AUTOMATION_TEST like so to specify custom base:

IMPLEMENT_CUSTOM_SIMPLE_AUTOMATION_TEST(TestMovement, FMyAutomationTest, "Blarg.Blarg.TestMovement",
								 EAutomationTestFlags::EditorContext | EAutomationTestFlags::EngineFilter)