small update for tests