Fixed another missing unit test