Fix import
Merge pull request #192 from mmun/fix-import

Fix import