Fixed import
Merge pull request #632 from sinflo/master

Fixed import