Fix import.
Merge pull request #493 from suraia/fix-import

Fix import