Fix PTL step calculation
Signed-off-by: smajumdar <smajumdar@nvidia.com>

# What does this PR do ?

Fixes an issue with step calculation logic.

**Collection**: [Core]

# Changelog 
- Update tests to increase num steps.

**PR Type**:
- [ ] New Feature
- [x] Bugfix
- [ ] Documentation

