diff --git a/store/postgres/postgres_test.go b/store/postgres/postgres_test.go
index 0cc3353..acb09b4 100644
--- a/store/postgres/postgres_test.go
+++ b/store/postgres/postgres_test.go
@@ -152,7 +152,7 @@ func TestCheckpoint_Get_NoRows(t *testing.T) {
 	if err := mock.ExpectationsWereMet(); err != nil {
 		t.Errorf("there were unfulfilled expectations: %s", err)
 	}
-	_ = ck.Shutdown()
+	_ = ck.Shutdown(ctx)
 }
 
 func TestCheckpoint_Get_QueryError(t *testing.T) {
diff --git a/store/redis/redis_test.go b/store/redis/redis_test.go
index da333c3..5a7aa0c 100644
--- a/store/redis/redis_test.go
+++ b/store/redis/redis_test.go
@@ -1,4 +1,4 @@
-// go:build unit
+//go:build unit
 
 package redis