Didn't see you used different class
Merge pull request #2 from huglester/patch-1

Didn't see you used different class