diff --git a/src/stratifier.c b/src/stratifier.c
index 8db001f6..7dfffbce 100644
--- a/src/stratifier.c
+++ b/src/stratifier.c
@@ -562,7 +562,7 @@ static void update_diff(ckpool_t *ckp)
 	json_decref(val);
 
 	ck_wlock(&workbase_lock);
-	proxy_base.diff = diff;
+	current_workbase->diff = proxy_base.diff = diff;
 	ck_wunlock(&workbase_lock);
 }