Tweak the _main shim for test.swift