Fixed "class" error