fix: fix broadcast bug in getp

[AayushSabharwal]

Checklist

• [ ] Appropriate tests were added
• [ ] Any code changes were done in a way that does not break public API
• [ ] All documentation related to code changes were updated
• [ ] The new code follows the contributor guidelines, in particular the SciML Style Guide and COLPRAC.
• [ ] Any new documentation only uses public API

Additional context

Add any other context about the problem here.

[codecov[bot]]

Codecov Report

All modified and coverable lines are covered by tests :white_check_mark:

Project coverage is 38.17%. Comparing base (2a45e9c) to head (6997781).

Additional details and impacted files

```diff @@ Coverage Diff @@ ## master #56 +/- ## ======================================= Coverage 38.17% 38.17% ======================================= Files 7 7 Lines 296 296 ======================================= Hits 113 113 Misses 183 183 ```

:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.