fixed a bug with the wrong imported class
Merge pull request #3 from znahas/master

fixed a bug with the wrong imported class