Pass machine name instead of class name
Merge pull request #1 from Takki/master

Pass machine name instead of class name