fix DM tests