diff --git a/drivers/of/unittest.c b/drivers/of/unittest.c
index 872956500c276382444f713b64b0c21596132cfe..66037511f2d707ae0d0f2392cfcf5e607500c4df 100644
--- a/drivers/of/unittest.c
+++ b/drivers/of/unittest.c
@@ -2521,6 +2521,10 @@ static int __init of_unittest(void)
 	int res;
 
 	/* adding data for unittest */
+
+	if (IS_ENABLED(CONFIG_UML))
+		unittest_unflatten_overlay_base();
+
 	res = unittest_data_add();
 	if (res)
 		return res;