Adding a missing import for previous commit.
Merge pull request #7 from mesyelp/master

Adding a missing import for previous commit.