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]

Codecov Report

All modified and coverable lines are covered by tests ✅

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

Additional details and impacted files

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

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.