Fixing bad import
Merge pull request #12 from IlianIliev/master

Bad import fix