X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/06c9ea02435fb56e8c32a31694a414ca161dc2ce..e73f90ddd2252f2b21d573ad2179137dca5dcd97:/services/keep-balance/collection_test.go

diff --git a/services/keep-balance/collection_test.go b/services/keep-balance/collection_test.go
index e6925c4afd..3ab9d07b2e 100644
--- a/services/keep-balance/collection_test.go
+++ b/services/keep-balance/collection_test.go
@@ -5,10 +5,11 @@
 package main
 
 import (
+	"context"
 	"sync"
 	"time"
 
-	"git.curoverse.com/arvados.git/sdk/go/arvados"
+	"git.arvados.org/arvados.git/sdk/go/arvados"
 	check "gopkg.in/check.v1"
 )
 
@@ -29,7 +30,7 @@ func (s *integrationSuite) TestIdenticalTimestamps(c *check.C) {
 			longestStreak := 0
 			var lastMod time.Time
 			sawUUID := make(map[string]bool)
-			err := EachCollection(s.client, pageSize, func(c arvados.Collection) error {
+			err := EachCollection(context.Background(), s.client, pageSize, func(c arvados.Collection) error {
 				if c.ModifiedAt.IsZero() {
 					return nil
 				}