Update TU shim for tests.