Fixes #4351: fixing import
Merge pull request #413 from VinceMacBuche/bug/4351/fixing_import

Fixes #4351: fixing import