diff --git a/mfs_test.go b/mfs_test.go
index dde5562..539b71a 100644
--- a/mfs_test.go
+++ b/mfs_test.go
@@ -817,19 +817,19 @@ func TestFlushing(t *testing.T) {
 		t.Fatal(err)
 	}
 
-	if err := FlushPath(rt, "/a/b/c/TEST"); err != nil {
+	if err := FlushPath(ctx, rt, "/a/b/c/TEST"); err != nil {
 		t.Fatal(err)
 	}
 
-	if err := FlushPath(rt, "/a/b/d/TEST"); err != nil {
+	if err := FlushPath(ctx, rt, "/a/b/d/TEST"); err != nil {
 		t.Fatal(err)
 	}
 
-	if err := FlushPath(rt, "/a/b/e/TEST"); err != nil {
+	if err := FlushPath(ctx, rt, "/a/b/e/TEST"); err != nil {
 		t.Fatal(err)
 	}
 
-	if err := FlushPath(rt, "/FILE"); err != nil {
+	if err := FlushPath(ctx, rt, "/FILE"); err != nil {
 		t.Fatal(err)
 	}
 
diff --git a/ops.go b/ops.go
index d989bb5..6e99e23 100644
--- a/ops.go
+++ b/ops.go
@@ -226,7 +226,7 @@ func DirLookup(d *Directory, pth string) (FSNode, error) {
 
 // TODO: Document this function and link its functionality
 // with the republisher.
-func FlushPath(rt *Root, pth string) error {
+func FlushPath(ctx context.Context, rt *Root, pth string) error {
 	nd, err := Lookup(rt, pth)
 	if err != nil {
 		return err
@@ -237,6 +237,6 @@ func FlushPath(rt *Root, pth string) error {
 		return err
 	}
 
-	rt.repub.WaitPub(context.TODO())
+	rt.repub.WaitPub(ctx)
 	return nil
 }