Fixed broken imports
Merge pull request #2 from mpharrigan/fix

Fixed broken imports