Test import fix

[codecov-commenter]

Codecov Report

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

Project coverage is 81.31%. Comparing base (9a8ea26) to head (2037832).

Additional details and impacted files

```diff @@ Coverage Diff @@ ## main #85 +/- ## =========================================== + Coverage 34.29% 81.31% +47.01% =========================================== Files 171 171 Lines 6557 6557 =========================================== + Hits 2249 5332 +3083 + Misses 4308 1225 -3083 ``` | [Flag](https://app.codecov.io/gh/facebookresearch/Pearl/pull/85/flags?src=pr&el=flags&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=facebookresearch) | Coverage Δ | | |---|---|---| | [unittests](https://app.codecov.io/gh/facebookresearch/Pearl/pull/85/flags?src=pr&el=flag&utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=facebookresearch) | `81.31% <100.00%> (+47.01%)` | :arrow_up: | Flags with carried forward coverage won't be shown. [Click here](https://docs.codecov.io/docs/carryforward-flags?utm_medium=referral&utm_source=github&utm_content=comment&utm_campaign=pr+comments&utm_term=facebookresearch#carryforward-flags-in-the-pull-request-comment) to find out more.

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