Fixed class not found error