Move method into TestRunContextExtensions