Fixing class reference.