small updates for tests