[BUGFIX] Remove remaining reference to removed class
Merge pull request #28 from NamelessCoder/rebase

[BUGFIX] Remove remaining reference to removed class