Fix import statement
Merge pull request #13 from dougwt/master

Fix import statement