fixing bad import
Merge pull request #6 from bowilliams/master

fixing bad import