Fix import
Merge pull request #1427 from psav/fix_ci_test

Fix import