Fix test.

3 jobs for enhancement/migrate_to_pytest in 4 minutes and 57 seconds (queued for 8 seconds)