Fix tests import error

[codecov-commenter]

Codecov Report

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

Comparison is base (e438278) 73.96% compared to head (c5f5e90) 73.96%.

Additional details and impacted files

```diff @@ Coverage Diff @@ ## master #171 +/- ## ======================================= Coverage 73.96% 73.96% ======================================= Files 38 38 Lines 2385 2385 ======================================= Hits 1764 1764 Misses 621 621 ```

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