fix testMain in TestDriver