FIX DI issue
Merge pull request #6 from aveenaash/master

FIX DI issue