Fixing import error
Merge pull request #29 from sandip-d/test_parallelism

Fixing import error