diff --git a/examples/db/migrations/20202806225100_create_todos.go b/examples/db/migrations/20202806225100_create_todos.go
index d4d4387..eed8b82 100644
--- a/examples/db/migrations/20202806225100_create_todos.go
+++ b/examples/db/migrations/20202806225100_create_todos.go
@@ -21,9 +21,9 @@ func MigrateCreateTodos(schema *rel.Schema) {
 
 	schema.CreateIndex("todos", "order", []string{"order"})
 
-	schema.Do(func(repo rel.Repository) error {
+	schema.Do(func(ctx context.Context, repo rel.Repository) error {
 		// add seeds
-		return repo.Insert(context.TODO(), &Todo{Title: "Do homework"})
+		return repo.Insert(ctx, &Todo{Title: "Do homework"})
 	})
 }
 
diff --git a/go.mod b/go.mod
index ca9c150..b8fb5d1 100644
--- a/go.mod
+++ b/go.mod
@@ -3,9 +3,9 @@ module github.com/go-rel/doc
 go 1.14
 
 require (
-	github.com/go-rel/migration v0.2.1
+	github.com/go-rel/migration v0.3.1
 	github.com/go-rel/mysql v0.7.0
-	github.com/go-rel/rel v0.35.0
+	github.com/go-rel/rel v0.36.0
 	github.com/go-rel/reltest v0.9.0
 	github.com/go-sql-driver/mysql v1.6.0
 	github.com/kr/pretty v0.1.0 // indirect
diff --git a/go.sum b/go.sum
index dfc6fb5..fe10ab5 100644
--- a/go.sum
+++ b/go.sum
@@ -4,17 +4,16 @@ github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSs
 github.com/fsnotify/fsnotify v1.4.7/go.mod h1:jwhsz4b93w/PPRr/qN1Yymfu8t87LnFCMoQvtojpjFo=
 github.com/fsnotify/fsnotify v1.4.9 h1:hsms1Qyu0jgnwNXIxa+/V/PDsU6CfLf6CNO8H7IWoS4=
 github.com/fsnotify/fsnotify v1.4.9/go.mod h1:znqG4EE+3YCdAaPaxE2ZRY/06pZUdp0tY4IgpuI1SZQ=
-github.com/go-rel/migration v0.2.1 h1:7msAEboglLZVKXsMuQ3kszfWPKAVR6sXnReVUOyEXrY=
-github.com/go-rel/migration v0.2.1/go.mod h1:/HWr0eHa+Pk99mQrDiSkTy5l+WNQDAmIJ4sbYOCEdzY=
+github.com/go-rel/migration v0.3.1 h1:Y9wSeBOsaZaSiiTchwS1GBhvLxQMnnOSSLAjgfSSckc=
+github.com/go-rel/migration v0.3.1/go.mod h1:qIfXgsIK61xUT4YKdt/ceQ5mTvzOfNa3Z69dcrAcRKw=
 github.com/go-rel/mysql v0.7.0 h1:6nYiGZ3HpHFHRpIeQgqXMcSf/ix3gigzg1/T7B+fXO0=
 github.com/go-rel/mysql v0.7.0/go.mod h1:eGyAntAR6O7LoSFc8u3MhVcC1s4VVwyogWZCRlojUcM=
 github.com/go-rel/primaryreplica v0.4.0 h1:lhU+4dh0/sDQEs602Chiz0SJDXewPU06baWQlx7oB3c=
 github.com/go-rel/primaryreplica v0.4.0/go.mod h1:HUBz+BUvUcg9JpRRk9PstV9J/qlEOqaHIIllGncbUs8=
 github.com/go-rel/rel v0.33.1/go.mod h1:DhB9Xmt/ymaumJAB6Z3Kq+IybLwQhhrzr6ZLeXMygPo=
 github.com/go-rel/rel v0.34.0/go.mod h1:YDHtnJhlaiSjANLGiZXvI57jg1amfpJSJlOimq6WJTE=
-github.com/go-rel/rel v0.35.0 h1:A+jyWX8m4PSkXeFC22zq1hUXoLs/tXeP8Oitc3ZMqTg=
-github.com/go-rel/rel v0.35.0/go.mod h1:mXRZv7sVwtc/ztcY6dgzE5RV5MC+4/2PHPQUBAGiUG8=
-github.com/go-rel/reltest v0.8.1/go.mod h1:ZGM+yKATZ1y46l3alApuUz9wu5o3twAmPfCVvJBjpF8=
+github.com/go-rel/rel v0.36.0 h1:GgKB+0btoZVQ+6YBboPJvRksNtbMJOTv+iMO79t+QMo=
+github.com/go-rel/rel v0.36.0/go.mod h1:SHjO9VOInj8GD2cTVWmTnOS+rj5IJZoDy8LEyd5oDUQ=
 github.com/go-rel/reltest v0.9.0 h1:OTBY2dD/hNXaH5r1XXIHRdni6QGCmFaQm1pFAFkg1ok=
 github.com/go-rel/reltest v0.9.0/go.mod h1:Eu1JNASF6TfiktPlmvWEYHp+UejmGl9MmKRWieHxkXo=
 github.com/go-rel/sql v0.9.1 h1:2tHu4U4Bk1DWKvzsitwcvv1sk6aGcxPh3LYJXspgk8w=
@@ -65,7 +64,7 @@ github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/
 github.com/stretchr/testify v1.7.4 h1:wZRexSlwd7ZXfKINDLsO4r7WBt3gTKONc6K/VesHvHM=
 github.com/stretchr/testify v1.7.4/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU=
 github.com/subosito/gotenv v1.2.0/go.mod h1:N0PQaV/YGNqwC0u51sEeR/aUtSLEXKX9iv69rRypqCw=
-github.com/subosito/gotenv v1.3.0/go.mod h1:YzJjq/33h7nrwdY+iHMhEOEEbW0ovIz0tB6t6PwAXzs=
+github.com/subosito/gotenv v1.4.0/go.mod h1:mZd6rFysKEcUhUHXJk0C/08wAgyDBFuwEYL7vWWGaGo=
 github.com/yuin/goldmark v1.2.1/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74=
 golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w=
 golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI=
@@ -127,6 +126,5 @@ gopkg.in/yaml.v2 v2.3.0/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI=
 gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
 gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
 gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
-gopkg.in/yaml.v3 v3.0.0/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
 gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
 gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=