adding retry for tests