Update for some tests