Fix "class not found"