Remove unused import
Fix for failed test on 1.509
Fixed for all versions
Merge pull request #64 from rahulsom/master

Fix for failed test on 1.509