Fix argument issue in revised unit tests.