[Fix] Controller class detection
Merge pull request #6 from widop/fix-controller-class-detection

[Fix] Controller class detection