1 files changed, 3 insertions, 1 deletions
diff --git a/mgmt.c b/mgmt.c
index 4b8ef8b..aaf96a5 100644
--- a/mgmt.c
+++ b/mgmt.c
@@ -34,7 +34,9 @@ static void mgmt_digest_step(struct mog_fd *mfd)
          assert(mgmt->wbuf == NULL && "wbuf should be NULL here");
  
          switch (next) {
-        case MOG_DIGEST_CONTINUE: return;
+        case MOG_DIGEST_CONTINUE:
+        case MOG_DIGEST_YIELD:
+                return;
          case MOG_DIGEST_EOF:
                  mog_mgmt_fn_digest_emit(mgmt);
                  break;