Adjust for tests