fix: incorrect batch depth

[swarmHaseeb]

Checklist

• [ ] I have read the coding guide
• [ ] My change requires a documentation update and I have done it
• [ ] I have added tests to cover my changes.

Description

This PR fixes #3092 and enables the change in depth after diluting

Motivation and context (Optional)

Related Issue (Optional)

Screenshots (if appropriate):

---

This change is