Updates for tests