Fixed obsolete class reference
Merge pull request #9 from InebriatedSquirrel/iss01

Fixed obsolete class reference