Fix missing import

This fixes #17693
Merge pull request #17694 from makinacorpus/8

Fix missing import