Restore DML Code

Signed-off-by: slanau <silu@gft.com>