Fix failing test units.