Unit test update